diff --git a/.github/workflows/python.yml b/.github/workflows/python.yml
index 09c27756be..22b40d484f 100644
--- a/.github/workflows/python.yml
+++ b/.github/workflows/python.yml
@@ -28,11 +28,6 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install setuptools wheel pytest
-          
-      - name: Run Unit Tests
-        working-directory: ./photon-lib/py
-        run: |
-          pytest
 
       - name: Build wheel
         working-directory: ./photon-lib/py
@@ -40,6 +35,13 @@ jobs:
           git fetch --tags --force
           python setup.py sdist bdist_wheel
 
+      - name: Run Unit Tests
+        working-directory: ./photon-lib/py
+        run: |
+            pip install dist/photonlibpy-0.0.2-py3-none-any.whl
+            pytest
+  
+
       - name: Upload artifacts
         uses: actions/upload-artifact@master
         with: