diff --git a/.github/workflows/release_build.yml b/.github/workflows/release_build.yml
index a6a2ca4f3..ebaacefc9 100644
--- a/.github/workflows/release_build.yml
+++ b/.github/workflows/release_build.yml
@@ -12,26 +12,25 @@ on:
 
 jobs:
   build_release:
-    runs-on: ${{ matrix.os }}
     strategy:
       matrix:
         include:
-         - os: linux.g5.12xlarge.nvidia.gpu
-           python-version: 3.9
-           python-tag: "py39"
-           cuda-tag: "cu121"
-         - os: linux.g5.12xlarge.nvidia.gpu
-           python-version: '3.10'
-           python-tag: "py310"
-           cuda-tag: "cu121"
-         - os: linux.g5.12xlarge.nvidia.gpu
-           python-version: '3.11'
-           python-tag: "py311"
-           cuda-tag: "cu121"
-         - os: linux.g5.12xlarge.nvidia.gpu
-           python-version: '3.12'
-           python-tag: "py312"
-           cuda-tag: "cu121"
+          - os: linux.g5.12xlarge.nvidia.gpu
+            python-version: 3.9
+            python-tag: "py39"
+            cuda-tag: "cu121"
+          - os: linux.g5.12xlarge.nvidia.gpu
+            python-version: '3.10'
+            python-tag: "py310"
+            cuda-tag: "cu121"
+          - os: linux.g5.12xlarge.nvidia.gpu
+            python-version: '3.11'
+            python-tag: "py311"
+            cuda-tag: "cu121"
+          - os: linux.g5.12xlarge.nvidia.gpu
+            python-version: '3.12'
+            python-tag: "py312"
+            cuda-tag: "cu121"
     uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
     with:
       runner: ${{ matrix.os }}
@@ -66,6 +65,6 @@ jobs:
             --username __token__ \
             --password "$PYPI_TOKEN" \
             --skip-existing \
-            torchrec-*.whl
+            dist/torchrec-*.whl