diff --git a/.github/workflows/publish_base_image.yml b/.github/workflows/publish_base_image.yml
index afaad9c1..26c82726 100644
--- a/.github/workflows/publish_base_image.yml
+++ b/.github/workflows/publish_base_image.yml
@@ -32,6 +32,7 @@ jobs:
             CUDA_VERSION=12.1
           tags: |
             vectorchai/scalellm_builder:cuda12.1-ubuntu22.04
+            vectorchai/scalellm_builder:cuda12.1
 
       - name: Build base for cuda 11.8
         uses: docker/build-push-action@v5
@@ -44,4 +45,5 @@ jobs:
             CUDA_VERSION=11.8
           tags: |
             vectorchai/scalellm_builder:cuda11.8-ubuntu22.04
+            vectorchai/scalellm_builder:cuda11.8