diff --git a/.github/workflows/pypi.yml b/.github/workflows/pypi.yml
index 5a02d4db..1cfb2df9 100644
--- a/.github/workflows/pypi.yml
+++ b/.github/workflows/pypi.yml
@@ -30,16 +30,16 @@ jobs:
             venv-sdist/bin/python -m pip install numpy
             venv-sdist/bin/python -m pip install ../dist/pymc_extras*.tar.gz
             echo "Checking import and version number (on release)"
-            venv-sdist/bin/python -c "import pymc as pmx; assert pmx.__version__ == '${{ github.ref_name }}'[1:] if '${{ github.ref_type }}' == 'tag' else True; print(pmx.__version__)"
+            venv-sdist/bin/python -c "import pymc_extras as pmx; assert pmx.__version__ == '${{ github.ref_name }}'[1:] if '${{ github.ref_type }}' == 'tag' else True; print(pmx.__version__)"
             cd ..
       - name: Check the bdist installs and imports
         run: |
               mkdir -p test-bdist
               cd test-bdist
               python -m venv venv-bdist
-              venv-bdist/bin/python -m pip install ../dist/pymc*.whl
+              venv-bdist/bin/python -m pip install ../dist/pymc_extras*.whl
               echo "Checking import and version number (on release)"
-              venv-bdist/bin/python -c "import pymc as pmx; assert pmx.__version__ == '${{ github.ref_name }}'[1:] if '${{ github.ref_type }}' == 'tag' else pmx.__version__; print(pmx.__version__)"
+              venv-bdist/bin/python -c "import pymc_extras as pmx; assert pmx.__version__ == '${{ github.ref_name }}'[1:] if '${{ github.ref_type }}' == 'tag' else pmx.__version__; print(pmx.__version__)"
               cd ..
       - uses: actions/upload-artifact@v3
         with: