diff --git a/requirements_mtj.txt b/requirements_mtj.txt
index d9b403eb..ad205326 100644
--- a/requirements_mtj.txt
+++ b/requirements_mtj.txt
@@ -2,7 +2,7 @@ numpy
 tqdm
 requests
 optax >= 0.0.5, <= 0.0.9
-dm-haiku
+dm-haiku == 0.0.5
 ray[default]
 jax == 0.2.21
 transformers
@@ -15,4 +15,4 @@ flask-ngrok
 eventlet
 lupa==1.10
 markdown
-bleach
\ No newline at end of file
+bleach