diff --git a/.github/workflows/linux_job_v2.yml b/.github/workflows/linux_job_v2.yml
index 595af0cffa..62719524a4 100644
--- a/.github/workflows/linux_job_v2.yml
+++ b/.github/workflows/linux_job_v2.yml
@@ -165,7 +165,7 @@ jobs:
         if: ${{ inputs.gpu-arch-type != 'rocm' }}
 
       - name: Setup ROCM
-        uses: pytorch/pytorch/.github/actions/setup-rocm@main
+        uses: pytorch/pytorch/.github/actions/setup-rocm@move_aws_steps_inside_setup_rocm
         if: ${{ inputs.gpu-arch-type == 'rocm' }}
 
       - name: Setup SSH