diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 2c2299d..12baaf7 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -29,7 +29,7 @@ jobs:
       run: |
         sudo apt-get install -y libopenmpi-dev
         python -m pip install --upgrade pip
-        pip install jax=0.4.35
+        pip install jax==0.4.35
         pip install .[test]
 
     - name: Run Single Device Tests