diff --git a/.github/workflows/build-test-publish-wheel.yml b/.github/workflows/build-test-publish-wheel.yml
index d109b74..cc70317 100644
--- a/.github/workflows/build-test-publish-wheel.yml
+++ b/.github/workflows/build-test-publish-wheel.yml
@@ -14,7 +14,7 @@
 
 name: Build, test, and publish a PyPi wheel (to testpypi)
 
-on: 
+on:
   push:
     branches:
       - main
@@ -28,22 +28,19 @@ jobs:
   build-test-publish-wheel:
     uses: NVIDIA/NeMo-FW-CI-templates/.github/workflows/_build_test_publish_wheel.yml@v0.12.2
     with:
-      image-name: nemo_container
+      image-name: nemo_run_container
       dockerfile: Dockerfile
-      image-label: nemo-aligner
+      image-label: nemo-run
       build-args: |
-        IMAGE_LABEL=nemo-aligner
-        MAX_JOBS=32
-        ALIGNER_COMMIT=${{ github.event.pull_request.head.sha || github.sha }}
+        IMAGE_LABEL=nemo-run
+        NEMO_RUN_COMMIT=${{ github.event.pull_request.head.sha || github.sha }}
       prune-filter-timerange: 24h
       dry-run: true
-      python-package: nemo_aligner
-      container-workdir: /opt/NeMo-Aligner
+      python-package: nemo_run/src
+      container-workdir: /opt/NeMo-Run
       environment: public
     secrets:
       TWINE_USERNAME: ${{ secrets.TWINE_USERNAME }}
       TWINE_PASSWORD: ${{ secrets.TWINE_PASSWORD }}
       SLACK_WEBHOOK_ADMIN: ${{ secrets.SLACK_WEBHOOK_ADMIN }}
       SLACK_WEBHOOK: ${{ secrets.SLACK_WEBHOOK }}
-
-  
\ No newline at end of file
diff --git a/Dockerfile b/Dockerfile
index 486b45d..f2ffd35 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,7 +10,7 @@ git init
 git remote add origin https://github.com/NVIDIA/NeMo-Aligner.git
 git fetch --all
 git fetch origin '+refs/pull/*/merge:refs/remotes/pull/*/merge'
-git checkout $CURATOR_COMMIT
+git checkout $NEMO_RUN_COMMIT
 EOF
 
 FROM python:3.12