diff --git a/.github/workflows/MLPipelinePostMerge.yml b/.github/workflows/MLPipelinePostMerge.yml
index edcf24d..a83a6b4 100644
--- a/.github/workflows/MLPipelinePostMerge.yml
+++ b/.github/workflows/MLPipelinePostMerge.yml
@@ -44,11 +44,11 @@ jobs:
         dvc push -r dev
 
 
-  Register-New-Model:
+  Register-New-Model-Dev:
     if: github.event.pull_request.merged == true
     runs-on: ubuntu-latest
     steps:
-    - uses: actions/checkout@v3
+    - uses: actions/checkout@v4
       with:
         fetch-depth: 0
 
@@ -68,11 +68,7 @@ jobs:
         REGISTER_MODEL_NAME=$(echo ${{ github.event.pull_request.head.ref }} | awk -F"-" '{print $1}')
         # REGISTER_MODEL_NAME=$(echo ${{github.ref_name}} | awk -F"-" '{print $1}')
         git config user.name "Github-Bot"
-        git config user.email "Github-Bot"
-
-        git fetch origin
-        git merge origin/master
-        git pull
+        git config user.email "Github-Bot@no-reply.com"
 
         gto register ${REGISTER_MODEL_NAME}
         gto assign ${REGISTER_MODEL_NAME} --stage dev
@@ -80,6 +76,7 @@ jobs:
 
   Register-Prediction-Image-Dev:
     needs: Promote-Model-To-Dev
+    # needs: [Promote-Model-To-Dev, Register-New-Model-Dev] WILL ADD BACK ONCE REGISTER WORKS
     runs-on: ubuntu-latest
 
     steps:
@@ -100,10 +97,11 @@ jobs:
         cd modules/ml-pipeline/src/pipeline/src
         dvc pull -r dev
 
-    - name: Build Prediction docker image (TODO - NEED LAMBDA IMAGE)
+    - name: Build Prediction docker image (TODO - NEED LAMBDA IMAGE, need to add version from gto registry)
       run: |
         cd modules/ml-pipeline/src/pipeline/
-        docker build . --file Prediction.Dockerfile --tag prediction_test
+        REGISTER_MODEL_NAME=$(echo ${{ github.event.pull_request.head.ref }} | awk -F"-" '{print $1}')
+        docker build . --file Prediction.Dockerfile --tag ${REGISTER_MODEL_NAME}
 
     - name: ECR Login - Dev
       env: