diff --git a/.github/workflows/build-and-push-image.yaml b/.github/workflows/build-and-push-image.yaml
index 7c61881..1a749dd 100644
--- a/.github/workflows/build-and-push-image.yaml
+++ b/.github/workflows/build-and-push-image.yaml
@@ -51,7 +51,7 @@ jobs:
   build_finetune_image:
     name: Build and Push llm-finetune image
     needs: set_version_tags
-    uses: truefoundry/workflows/.github/workflows/build.yml@free-disk-space-input
+    uses: truefoundry/workflows/.github/workflows/build.yml@v0.1.1-rc.1
     with:
       image_tag: ${{ github.sha }}
       extra_image_tag: |
@@ -68,7 +68,7 @@ jobs:
 
   build_jupyter_image:
     name: Build and Push jupyter image
-    uses: truefoundry/workflows/.github/workflows/build.yml@free-disk-space-input
+    uses: truefoundry/workflows/.github/workflows/build.yml@v0.1.1-rc.1
     with:
       image_tag: ${{ needs.set_version_tags.outputs.jupyter_version_tag }}
       platforms: linux/amd64