diff --git a/.github/workflows/image_build.yml b/.github/workflows/image_build.yml
index 22f72131..8f1386c6 100644
--- a/.github/workflows/image_build.yml
+++ b/.github/workflows/image_build.yml
@@ -227,7 +227,7 @@ jobs:
         with:
           builder: ${{ steps.buildx.outputs.name }}
           build-args: |
-            BASE_IMAGE=localai/localai:${{ steps.meta_aio_dockerhub.outputs.version }}
+            BASE_IMAGE=localai/localai:${{ steps.meta.outputs.version }}
           context: .
           file: ./Dockerfile.aio
           platforms: ${{ inputs.platforms }}
@@ -240,4 +240,4 @@ jobs:
       - name: job summary(AIO)
         if: inputs.aio != ''
         run: |
-          echo "Built image: ${{ steps.meta_aio.outputs.labels }}" >> $GITHUB_STEP_SUMMARY
\ No newline at end of file
+          echo "Built image: ${{ steps.meta_aio.outputs.labels }}" >> $GITHUB_STEP_SUMMARY