diff --git a/metaworld-jax/requirements.txt b/metaworld-jax/requirements.txt
index bba6a3a..0be7581 100644
--- a/metaworld-jax/requirements.txt
+++ b/metaworld-jax/requirements.txt
@@ -16,4 +16,4 @@ orbax-checkpoint==0.7.0
 tyro==0.8.11
 
 # Algorithms
-mtrl @ git+https://github.com/rainx0r/mtrl@9cf5a39
+mtrl @ git+https://github.com/rainx0r/mtrl@a3a44a0