diff --git a/unsloth/__init__.py b/unsloth/__init__.py
index bbe728045c..068c9651df 100644
--- a/unsloth/__init__.py
+++ b/unsloth/__init__.py
@@ -267,6 +267,7 @@ from .trainer import *
 from unsloth_zoo.rl_environments import (
     check_python_modules,
     create_locked_down_function,
+    execute_with_time_limit,
     Benchmarker,
 )
 
diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index 328049af20..902d3ee6a8 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -94,6 +94,7 @@ from unsloth_zoo.tokenizer_utils import (
 from unsloth_zoo.rl_environments import (
     check_python_modules,
     create_locked_down_function,
+    execute_with_time_limit,
     Benchmarker,
 )
 from unsloth_zoo.patching_utils import (