diff --git a/fancy_gym/__init__.py b/fancy_gym/__init__.py
index f6f690a..32308fa 100644
--- a/fancy_gym/__init__.py
+++ b/fancy_gym/__init__.py
@@ -1,5 +1,6 @@
 from fancy_gym import dmc, meta, open_ai
-from fancy_gym.utils.make_env_helpers import make, make_bb, make_rank
+from fancy_gym.utils.make_env_helpers import make_bb
+from .envs.registry import register
 from .dmc import ALL_DMC_MOVEMENT_PRIMITIVE_ENVIRONMENTS
 # Convenience function for all MP environments
 from .envs import ALL_FANCY_MOVEMENT_PRIMITIVE_ENVIRONMENTS
@@ -7,7 +8,5 @@ from .meta import ALL_METAWORLD_MOVEMENT_PRIMITIVE_ENVIRONMENTS
 from .open_ai import ALL_GYM_MOVEMENT_PRIMITIVE_ENVIRONMENTS
 
 ALL_MOVEMENT_PRIMITIVE_ENVIRONMENTS = {
-    key: value + ALL_DMC_MOVEMENT_PRIMITIVE_ENVIRONMENTS[key] +
-         ALL_GYM_MOVEMENT_PRIMITIVE_ENVIRONMENTS[key] +
-         ALL_METAWORLD_MOVEMENT_PRIMITIVE_ENVIRONMENTS[key]
+    key: value + ALL_DMC_MOVEMENT_PRIMITIVE_ENVIRONMENTS[key] + ALL_GYM_MOVEMENT_PRIMITIVE_ENVIRONMENTS[key] + ALL_METAWORLD_MOVEMENT_PRIMITIVE_ENVIRONMENTS[key]
     for key, value in ALL_FANCY_MOVEMENT_PRIMITIVE_ENVIRONMENTS.items()}