diff --git a/tests/requirements.txt b/tests/requirements.txt
index 8ad9777..ebda1f6 100644
--- a/tests/requirements.txt
+++ b/tests/requirements.txt
@@ -12,4 +12,4 @@ mace-torch
 newtonnet @ git+https://github.com/THGLab/NewtonNet.git
 sella==2.3.4
 jax<=0.4.24
-jaxlib<=0.4.24
+jaxlib<=0.4.28