diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index e50bc4bb..459e9b80 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -154,13 +154,8 @@ jobs:
           echo "runs_on=${{ env.RUNS_ON }}" >> "$GITHUB_OUTPUT"
           echo "label=${{ env.LABEL }}" >> "$GITHUB_OUTPUT"
   integration_tests:
-    concurrency:
-      group: ${{ github.workflow }}-${{ github.job }}-${{ needs.build-and-push.outputs.label }}-${{ github.head_ref || github.run_id }}
-      cancel-in-progress: true
-    needs: build-and-push
     runs-on:
-      group: ${{ needs.build-and-push.outputs.runs_on }}
-    if: needs.build-and-push.outputs.runs_on != 'ubuntu-latest'
+      group: aws-use1-g5-12xlarge
     env:
       PYTEST_FLAGS: ${{ (startsWith(github.ref, 'refs/tags/') || github.ref == 'refs/heads/main' || inputs.release-tests == true) && '--release' || '' }}
     steps:
@@ -178,8 +173,7 @@ jobs:
       - name: Run tests
         run: |
           export DOCKER_VOLUME=/mnt/cache
-          export DOCKER_IMAGE=${{ needs.build-and-push.outputs.docker_image }}
-          export DOCKER_DEVICES=${{ needs.build-and-push.outputs.docker_devices }}
+          export DOCKER_IMAGE=registry-push.github-runners.huggingface.tech/api-inference/community/text-generation-inference:sha-c3e7bcc
           export HF_TOKEN=${{ secrets.HF_TOKEN }}
           echo $DOCKER_IMAGE
           pytest -x -s -vv integration-tests ${PYTEST_FLAGS}