diff --git a/evojax/task/brax_task.py b/evojax/task/brax_task.py
index 9ae88f65..8ee21834 100644
--- a/evojax/task/brax_task.py
+++ b/evojax/task/brax_task.py
@@ -61,7 +61,6 @@ class BraxTask(VectorizedTask):
     def __init__(self,
                  env_name: str,
                  max_steps: int = 1000,
-                 legacy_spring: bool = True,
                  bd_extractor: Optional[BDExtractor] = None,
                  test: bool = False):
         self.max_steps = max_steps
@@ -69,8 +68,7 @@ def __init__(self,
         self.test = test
         brax_env = create(
             env_name=env_name,
-            episode_length=max_steps,
-            legacy_spring=legacy_spring,
+            episode_length=max_steps
         )
         self.obs_shape = tuple([brax_env.observation_size, ])
         self.act_shape = tuple([brax_env.action_size, ])