diff --git a/poetry.lock b/poetry.lock
index 60f4e88..43ee693 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -3743,4 +3743,4 @@ testing = ["big-O", "jaraco.functools", "jaraco.itertools", "more-itertools", "p
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.10,<3.12"
-content-hash = "f38359fe4a510e44b5a3bb2f4133d597ba4d3318b4b0f54b36a13ca38b9df7eb"
+content-hash = "85834e91787aa0750c8010cf2374e9ef76bc91680960623311d27ddcd653e5d1"
diff --git a/pyproject.toml b/pyproject.toml
index 682b3eb..1631d8f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,7 +13,7 @@ keywords = ["jax", "rl", "ppo"]
 python = ">=3.10,<3.12"
 flax = "^0.8.2"
 distrax = "^0.1.5"
-jax-tqdm = "^0.2.0"
+jax-tqdm = "0.2.0"
 gymnax = "^0.0.8"
 scipy = ">=1.13.0"