diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 853e723..0656590 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -24,6 +24,7 @@ jobs:
           python -m pip install --upgrade pip
           pip install pytest
           if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
+          if [ -f setup.py ]; then pip install .; fi
           pip install --upgrade "jax[cpu]"
       - name: Test with pytest
         run: |