diff --git a/setup.py b/setup.py
index bd46a62cb2..6a49fa72ed 100755
--- a/setup.py
+++ b/setup.py
@@ -57,7 +57,7 @@
     'jax==0.4.3',
     'jaxlib==0.4.3',
     'chex',
-    'dm-haiku',
+    'dm-haiku==0.0.10',
     'flax',
     'optax',
     'rlax',