diff --git a/.github/workflows/pypi-gpu.yml b/.github/workflows/pypi-gpu.yml
index 22095e065..2300fd638 100644
--- a/.github/workflows/pypi-gpu.yml
+++ b/.github/workflows/pypi-gpu.yml
@@ -68,7 +68,7 @@ jobs:
         if: matrix.target == 'x86_64'
         run: |
           pip install ezkl-gpu --no-index --find-links dist --force-reinstall
-          python -c "import ezkl-gpu"
+          python -c "import ezkl_gpu"
 
       - name: Upload wheels
         uses: actions/upload-artifact@v3