diff --git a/.github/workflows/wheel.yaml b/.github/workflows/wheel.yaml
index 81e8df2a..8a307d6a 100644
--- a/.github/workflows/wheel.yaml
+++ b/.github/workflows/wheel.yaml
@@ -39,19 +39,19 @@ jobs:
     strategy:
       fail-fast: false
     name: Build and Upload wheel
-    uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
+    uses: nicolashug/test-infra/.github/workflows/build_wheels_linux.yml@sdistsss
     with:
       repository: pytorch/torchcodec
       ref: ""
-      test-infra-repository: pytorch/test-infra
-      test-infra-ref: main
+      test-infra-repository: nicolashug/test-infra
+      test-infra-ref: sdistsss
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       post-script: packaging/post_build_script.sh
       smoke-test-script: packaging/fake_smoke_test.py
       package-name: torchcodec
       trigger-event: ${{ github.event_name }}
       build-platform: "python-build-package"
-      build-command: "BUILD_AGAINST_ALL_FFMPEG_FROM_S3=1 python -m build --wheel -vvv --no-isolation"
+      build-command: "BUILD_AGAINST_ALL_FFMPEG_FROM_S3=1 python -m build -vvv --no-isolation"
 
   install-and-test:
     runs-on: ubuntu-latest