diff --git a/requirements/requirements.txt b/requirements/requirements.txt
index cec7a600..44fe1bf7 100644
--- a/requirements/requirements.txt
+++ b/requirements/requirements.txt
@@ -3,7 +3,7 @@ colorama
 distrax @ git+https://github.com/google-deepmind/distrax  # distrax release doesn't support jax > 0.4.13
 flashbax @ git+https://github.com/instadeepai/flashbax
 flax
-gymnax>=0.6.0
+gymnax>=0.0.6
 huggingface_hub
 hydra-core==1.3.2
 jax>=0.4.10