diff --git a/.github/workflows/build_vllm.yaml b/.github/workflows/build_vllm.yaml
index c83875ae1..8c27c51ad 100644
--- a/.github/workflows/build_vllm.yaml
+++ b/.github/workflows/build_vllm.yaml
@@ -13,14 +13,14 @@ permissions:
 jobs:
   build:
     name: forge-cu129-nightly
-    uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@812cf38d3d0d763e0578e695a004e6f51673e47c
+    uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
     strategy:
       fail-fast: false
     with:
       repository: meta-pytorch/forge
       ref: ""
       test-infra-repository: pytorch/test-infra
-      test-infra-ref: 812cf38d3d0d763e0578e695a004e6f51673e47c
+      test-infra-ref: main
       run-smoke-test: false
       wheel-nightly-policy: gha_workflow_build_preview_wheels_policy
       wheel-upload-path: whl/preview/forge
diff --git a/.github/workflows/build_wheels.yaml b/.github/workflows/build_wheels.yaml
index 994116171..aceb2540f 100644
--- a/.github/workflows/build_wheels.yaml
+++ b/.github/workflows/build_wheels.yaml
@@ -13,14 +13,14 @@ permissions:
 jobs:
   build:
     name: forge-cu129-nightly
-    uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@812cf38d3d0d763e0578e695a004e6f51673e47c
+    uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
     strategy:
       fail-fast: false
     with:
       repository: meta-pytorch/forge
       ref: ""
       test-infra-repository: pytorch/test-infra
-      test-infra-ref: 812cf38d3d0d763e0578e695a004e6f51673e47c
+      test-infra-ref: main
       run-smoke-test: false
       wheel-nightly-policy: gha_workflow_build_preview_wheels_policy
       wheel-upload-path: whl/preview/forge
@@ -48,4 +48,4 @@ jobs:
       pre-script: .github/packaging/pre_build_gpu.sh
       post-script: .github/packaging/post_build_script.sh
       trigger-event: ${{ github.event_name }}
-      build-platform: 'python-build-package'
\ No newline at end of file
+      build-platform: 'python-build-package'