diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 7ff7ded..a657e1a 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -27,7 +27,7 @@ jobs:
       fail-fast: false
       matrix:
         python-version: ['3.11', '3.12', '3.13', '3.14']
-        jax-version: ['0.10.0', '0.10.1', '0.10.2', '0.11.0', '0.11.1', '0.11.2', '0.12.0', 'nightly']
+        jax-version: ['0.10.0', '0.10.1', '0.10.2', '0.11.0', '0.11.1', '0.11.2', '0.12.0', '0.12.0.1', 'nightly']
         exclude:
           - python-version: '3.11'
             jax-version: '0.11.0'
@@ -37,6 +37,8 @@ jobs:
             jax-version: '0.11.2'
           - python-version: '3.11'
             jax-version: '0.12.0'
+          - python-version: '3.11'
+            jax-version: '0.12.0.1'
 
     runs-on: ubuntu-latest
     timeout-minutes: 30