diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index c194aafe..f60a2f59 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -32,9 +32,11 @@ jobs:
         with:
           python-version: '3.10'
       - uses: actions/checkout@v4
-      - name: Run tests
+      - name: install deps
         run: |
           make install-integration-tests
+      - name: Run tests
+        run: |
           export DOCKER_IMAGE=registry.internal.huggingface.tech/api-inference/community/text-generation-inference:2.0.4-rocm
           export HUGGING_FACE_HUB_TOKEN=${{ secrets.HUGGING_FACE_HUB_TOKEN }}
-          python -m pytest -s -vv integration-tests
+          python -m pytest -s -vv integration-tests/models/test_flash_gpt2.py