diff --git a/.github/workflows/publish-to-test-pypi.yml b/.github/workflows/publish-to-test-pypi.yml
index 28856883e..4cf47e304 100644
--- a/.github/workflows/publish-to-test-pypi.yml
+++ b/.github/workflows/publish-to-test-pypi.yml
@@ -55,7 +55,7 @@ jobs:
     - name: Update version for nightly
       run: |
         sed -i 's/version="\([^"]*\)"/version="\1${{ needs.trigger-client-build.outputs.version }}"/' setup.py
-        sed -i 's/llama-stack-client>=\([^"]*\)/llama-stack-client==\1${{ needs.trigger-client-build.outputs.version }}/' setup.py
+        sed -i 's/llama-stack-client>=\([^"]*\)/llama-stack-client==\1${{ needs.trigger-client-build.outputs.version }}/' requirements.txt
     - name: Set up Python
       uses: actions/setup-python@v5
       with: