diff --git a/.github/workflows/publish-to-test-pypi.yml b/.github/workflows/publish-to-test-pypi.yml
index 9aa682b79..d61398e25 100644
--- a/.github/workflows/publish-to-test-pypi.yml
+++ b/.github/workflows/publish-to-test-pypi.yml
@@ -64,6 +64,7 @@ jobs:
     - name: Update version for nightly
       run: |
         sed -i 's/version="\([^"]*\)"/version="\1${{ steps.version.outputs.version }}"/' setup.py
+        set -i 's/llama-stack-client>="\([^"]*\)"/llama-stack-client=="\1${{ steps.version.outputs.version }}" --index-url https://test.pypi.org/simple/'
     - name: Set up Python
       uses: actions/setup-python@v5
       with: