diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index 75e31fe..ad9c799 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -33,7 +33,7 @@ jobs:
         with:
             python-version: "3.8"
             test-script: |
-                python -m pip install pytest jax jaxlib typeguard
+                python -m pip install pytest beartype equinox jaxlib
                 cp -r ${{ github.workspace }}/test ./test
                 pytest
             pypi-token: ${{ secrets.pypi_token }}
diff --git a/.github/workflows/run_tests.yml b/.github/workflows/run_tests.yml
index ef99b24..56d0d23 100644
--- a/.github/workflows/run_tests.yml
+++ b/.github/workflows/run_tests.yml
@@ -42,7 +42,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install pytest wheel jaxlib beartype equinox
+          python -m pip install pytest wheel beartype equinox jaxlib
 
       - name: Checks with pre-commit
         uses: pre-commit/action@v2.0.3