diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index 2b13962f..d035b3a6 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -191,7 +191,7 @@ jobs:
         run: |
           export DOCKER_VOLUME=/mnt2/
           # export DOCKER_IMAGE=${{ needs.build-and-push.outputs.docker_image }}
-          export DOCKER_IMAGE=busybox
+          export DOCKER_IMAGE=registry.internal.huggingface.tech/api-inference/community/text-generation-inference:sha-318bbc7-rocm
           # export DOCKER_DEVICES=${{ needs.build-and-push.outputs.docker_devices }}
           export DOCKER_DEVICES=/dev/kfd,/dev/cri
           # export EXTRA_PYTEST="${{ needs.build-and-push.outputs.extra_pytest }}"