diff --git a/.github/workflows/release.yaml b/.github/workflows/release.yaml
index 771fab629..652a873a1 100644
--- a/.github/workflows/release.yaml
+++ b/.github/workflows/release.yaml
@@ -25,10 +25,14 @@ on:
         required: true
         default: true
         type: boolean
+      version-bump-branch:
+        description: Branch for version bump
+        required: true
+        type: string
           
 jobs:
   release:
-    uses: NVIDIA/NeMo-FW-CI-templates/.github/workflows/_release_library.yml@v0.18.1
+    uses: NVIDIA/NeMo-FW-CI-templates/.github/workflows/_release_library.yml@v0.18.3
     with:
       release-ref: ${{ inputs.release-ref }}
       image-name: nemo_aligner_container
@@ -42,6 +46,7 @@ jobs:
       container-workdir: /opt/NeMo-Aligner
       library-name: NeMo-Aligner
       dry-run: ${{ inputs.dry-run }}
+      version-bump-branch: ${{ inputs.version-bump-branch }}
     secrets:
       TWINE_USERNAME: ${{ secrets.TWINE_USERNAME }}
       TWINE_PASSWORD: ${{ secrets.TWINE_PASSWORD }}