diff --git a/metaworld/__init__.py b/metaworld/__init__.py
index a2c1e14c..62be8fab 100644
--- a/metaworld/__init__.py
+++ b/metaworld/__init__.py
@@ -197,7 +197,7 @@ def __init__(self, env_name, seed=None):
         cls = _env_dict.ALL_V3_ENVIRONMENTS[env_name]
         self._train_classes = OrderedDict([(env_name, cls)])
         self._test_classes = OrderedDict([(env_name, cls)])
-        args_kwargs = _env_dict.ML1_args_kwargs[env_name]
+        args_kwargs = _env_dict.ML1_ARGS_KWARGS[env_name]
 
         self._train_tasks = _make_tasks(
             self._train_classes, {env_name: args_kwargs}, _MT_OVERRIDE, seed=seed
@@ -286,7 +286,7 @@ def __init__(self, env_name, seed=None):
         cls = _env_dict.ALL_V3_ENVIRONMENTS[env_name]
         self._train_classes = OrderedDict([(env_name, cls)])
         self._test_classes = self._train_classes
-        args_kwargs = _env_dict.ML1_args_kwargs[env_name]
+        args_kwargs = _env_dict.ML1_ARGS_KWARGS[env_name]
 
         self._train_tasks = _make_tasks(
             self._train_classes, {env_name: args_kwargs}, _ML_OVERRIDE, seed=seed
diff --git a/metaworld/env_dict.py b/metaworld/env_dict.py
index 06a51776..0bbb945c 100644
--- a/metaworld/env_dict.py
+++ b/metaworld/env_dict.py
@@ -331,7 +331,7 @@ def initialize(env, seed=None, render_mode=None):
 ML1_V3 = _get_train_test_env_dict(
     list(ALL_V3_ENVIRONMENTS.keys()), list(ALL_V3_ENVIRONMENTS.keys())
 )
-ML1_args_kwargs = _get_args_kwargs(ALL_V3_ENVIRONMENTS, ML1_V3["train"])
+ML1_ARGS_KWARGS = _get_args_kwargs(ALL_V3_ENVIRONMENTS, ML1_V3["train"])
 
 ML10_V3 = _get_train_test_env_dict(
     train_env_names=[