diff --git a/pyproject.toml b/pyproject.toml
index c52810b..cdb99f8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -41,8 +41,8 @@ dependencies = [
     "torch",
     "dill",
     "imageio",
-    "jax=0.4.20",
-    "jaxlib=0.4.20",
+    "jax==0.4.20",
+    "jaxlib==0.4.20",
     "evosax==0.1.6"
 ]