diff --git a/tests/requirements.txt b/tests/requirements.txt
index 8ad9777..c35761f 100644
--- a/tests/requirements.txt
+++ b/tests/requirements.txt
@@ -11,5 +11,5 @@ torchaudio
 mace-torch
 newtonnet @ git+https://github.com/THGLab/NewtonNet.git
 sella==2.3.4
-jax<=0.4.24
+jax<=0.4.30
 jaxlib<=0.4.24