diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index 2df9b878d0..9ba9daba74 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -91,6 +91,11 @@ from unsloth_zoo.log import logger
 from unsloth_zoo.tokenizer_utils import (
     patch_tokenizer as _patch_tokenizer,
 )
+from unsloth_zoo.rl_environments import (
+    check_python_modules,
+    create_locked_down_function,
+    Benchmarker,
+)
 from unsloth_zoo.patching_utils import (
     patch_compiling_bitsandbytes,
     patch_layernorm,