diff --git a/.github/workflows/publish-to-docker.yml b/.github/workflows/publish-to-docker.yml
index f113e813b..83fb01c34 100644
--- a/.github/workflows/publish-to-docker.yml
+++ b/.github/workflows/publish-to-docker.yml
@@ -23,9 +23,11 @@ jobs:
           username: ${{ secrets.DOCKERHUB_USERNAME }}
           password: ${{ secrets.DOCKERHUB_TOKEN }}
 
-      - name: Custom build docker image
+      - name: Install llama-stack
         run: |
-            pip install --no-cache --index-url https://pypi.org/simple/ --extra-index-url https://test.pypi.org/simple/ llama-stack==0.0.63.dev20250114
             pip install -e .
-            TESTPYPI_VERSION=0.0.63.dev20250114 llama stack build --template ollama --image-type docker
-            docker images
+
+      - name: Build docker image
+        run: |
+          llama stack build --template ollama --image-type docker
+          docker images