diff --git a/.github/workflows/publish-to-test-pypi.yml b/.github/workflows/publish-to-test-pypi.yml
index ce41adf4f..a1bb1d4ad 100644
--- a/.github/workflows/publish-to-test-pypi.yml
+++ b/.github/workflows/publish-to-test-pypi.yml
@@ -64,7 +64,11 @@ jobs:
     - name: Update version for nightly
       run: |
         sed -i 's/version="\([^"]*\)"/version="\1${{ steps.version.outputs.version }}"/' setup.py
-        sed -i 's/llama-stack-client>="\([^"]*\)"/llama-stack-client=="\1${{ steps.version.outputs.version }}"/' requirements.txt
+        echo "Before requirements.txt modification:"
+        cat requirements.txt
+        sed -i 's/llama-stack-client\([>=]\+\)"\([^"]*\)"/llama-stack-client=="\2${{ steps.version.outputs.version }}"/' requirements.txt
+        echo "After requirements.txt modification:"
+        cat requirements.txt
     - name: Set up Python
       uses: actions/setup-python@v5
       with: