diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index a633812d..8fb4c1e9 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -210,6 +210,7 @@ jobs:
           export DOCKER_IMAGE=registry-us-east-1.prod.aws.ci.huggingface.tech/api-inference/community/text-generation-inference:sha-c55c43e
           export HF_TOKEN=${{ secrets.HF_TOKEN }}
           echo $DOCKER_IMAGE
+          docker pull $DOCKER_IMAGE
           pytest -x -s -vv integration-tests ${PYTEST_FLAGS}
       - name: Setup tmate session
         if: ${{ failure() }}