diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
index a946db918..ccfeed070 100644
--- a/.github/workflows/docker-build.yml
+++ b/.github/workflows/docker-build.yml
@@ -28,6 +28,7 @@ jobs:
     runs-on: ubuntu-latest
     outputs:
       tags: ${{ steps.set-vars.outputs.tags }}
+      file: ${{ steps.set-vars.outputs.file }}
     steps:
       - uses: actions/checkout@v4
       - name: Set Dockerfile and Tags
@@ -84,7 +85,7 @@ jobs:
   build_components:
     if: ${{ inputs.release_type == 'main' }}
     runs-on: ubuntu-latest
-    needs: build_base
+    needs: build
     strategy:
       matrix:
         component: [backend, frontend]
@@ -121,7 +122,7 @@ jobs:
     name: Restart HuggingFace Spaces
     if: ${{ inputs.release_type == 'main' }}
     runs-on: ubuntu-latest
-    needs: build_base
+    needs: build
     strategy:
       matrix:
         python-version: