diff --git a/unsloth/__init__.py b/unsloth/__init__.py
index 1be571b69b..bbe728045c 100644
--- a/unsloth/__init__.py
+++ b/unsloth/__init__.py
@@ -264,6 +264,11 @@ from .save import *
 from .chat_templates import *
 from .tokenizer_utils import *
 from .trainer import *
+from unsloth_zoo.rl_environments import (
+    check_python_modules,
+    create_locked_down_function,
+    Benchmarker,
+)
 
 # Patch TRL trainers for backwards compatibility
 _patch_trl_trainer()