diff --git a/.github/workflows/build-docker-images.yml b/.github/workflows/build-docker-images.yml
index 45fd82c21d2a..69afe4677a42 100644
--- a/.github/workflows/build-docker-images.yml
+++ b/.github/workflows/build-docker-images.yml
@@ -78,7 +78,7 @@ jobs:
         with:
           context: ./docker/transformers-all-latest-gpu
           build-args: |
-            REF=update_dockerfile
+            REF=main
             PYTORCH=2.8.0
             TORCHCODEC=0.7.0
             FLASH_ATTN=yes