diff --git a/.github/workflows/pypi_upload.yml b/.github/workflows/pypi_upload.yml
index a01ee7405..e0ac98ea0 100644
--- a/.github/workflows/pypi_upload.yml
+++ b/.github/workflows/pypi_upload.yml
@@ -150,7 +150,7 @@ jobs:
         name: Install guidance from wheels
       - run: pip install transformers torch
         name: Other installs
-      - run: python -c "import guidance; lm = guidance.models.Transformers('gpt2'); lm += '1,2,3,4,5,' + gen('num', max_tokens=5); assert lm['num'].startswith("6")"
+      - run: python -c "import guidance; lm = guidance.models.Transformers('gpt2'); lm += '1,2,3,4,5,' + guidance.gen('num', max_tokens=5); assert lm['num'].startswith("6")"
         name: Run smoke test
 
   publish_wheels:
@@ -163,7 +163,7 @@ jobs:
       - uses: actions/download-artifact@v4
         with:
           name: collected_dist_files
-          path: wheelhouse
+          path: dist
 
       - name: Publish package to PyPI
         uses: pypa/gh-action-pypi-publish@release/v1