diff --git a/.github/workflows/MLPipelinePostMerge.yml b/.github/workflows/MLPipelinePostMerge.yml
index 4988534..13ea9c1 100644
--- a/.github/workflows/MLPipelinePostMerge.yml
+++ b/.github/workflows/MLPipelinePostMerge.yml
@@ -187,23 +187,27 @@ jobs:
         git config user.name "Github-Bot"
         git config user.email "Github-Bot@no-reply.com"
 
-        gto assign ${REGISTER_MODEL_NAME} --stage dev
+        latest_dev_version=$(gto history ${REGISTER_MODEL_NAME} --asc --plain | awk '{print $NF}' | awk '/dev/')
+        if [ -z "${latest_dev_version}" ]; then
+          increment_version="1"
+        else
+          increment_version=$(echo ${latest_dev_version} | awk 'END{print}' | awk -F"#" '{print $3}' | awk '{$1++; print}')
+        fi
 
-        # latest_dev_version=$(gto history ${REGISTER_MODEL_NAME} --asc --plain | awk '{print $NF}' | awk '/dev/')
-        # if [ -z "${latest_dev_version}" ]; then
-        #   increment_version="1"
-        # else
-        #   increment_version=$(echo ${latest_dev_version} | awk 'END{print}' | awk -F"#" '{print $3}' | awk '{$1++; print}')
-        # fi
+        new_tag=${REGISTER_MODEL_NAME}#dev#${increment_version}
+        latest_version=$(gto show model@latest --ref | awk -F"@" '{print $2}')
 
-        # new_tag=${REGISTER_MODEL_NAME}#dev#${increment_version}
-        # latest_version=$(gto show model@latest --ref | awk -F"@" '{print $2}')
+        echo ${new_tag}
 
-        # echo ${new_tag}
+        commit_hash=$(gto history ${REGISTER_MODEL_NAME} --asc --plain | awk "/${latest_version}/" | awk '{print $(NF-1)}')
+        git checkout ${commit_hash}
 
-        # git pull #Get new model registry md file changes
-        # git tag -a ${new_tag} -m "Assigning stage dev to artifact temp version ${latest_version}"
-        # git push origin ${new_tag}
+        git pull #Get new model registry md file changes
+        git tag -a ${new_tag} -m "Assigning stage dev to artifact ${REGISTER_MODEL_NAME} version ${latest_version}"
+        git push origin ${new_tag}
+
+        git checkout master
+        git fetch --all
 
         gto show > MODEL_REGISTRY.md
         git add .