diff --git a/.github/workflows/trtllm_tests.yaml b/.github/workflows/trtllm_tests.yaml
index 2aff01fa..3edb6440 100644
--- a/.github/workflows/trtllm_tests.yaml
+++ b/.github/workflows/trtllm_tests.yaml
@@ -32,14 +32,14 @@ jobs:
     secrets: inherit
 
   run-tests:
-    needs: build-and-push
+    needs: build-backend
     concurrency:
       group: ${{ github.workflow }}-${{ github.job }}-trtllm-${{ github.head_ref || github.run_id }}
       cancel-in-progress: true
     runs-on:
       group: aws-g6-12xl-plus-priv-cache
     container:
-      image: ${{ needs.build-and-push.outputs.docker_image }}
+      image: ${{ needs.build-backend.outputs.docker_image }}
       credentials:
         username: ${{ secrets.REGISTRY_USERNAME }}
         password: ${{ secrets.REGISTRY_PASSWORD }}