diff --git a/unsloth/models/llama.py b/unsloth/models/llama.py
index a337472a3e..c50f65e4bd 100644
--- a/unsloth/models/llama.py
+++ b/unsloth/models/llama.py
@@ -2739,3 +2739,5 @@ class FastLlamaModel:
     pass
 pass
 
+from .rl import PatchFastRL
+PatchFastRL(FastLanguageModel = FastLlamaModel)
diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py
index e89e657fab..31a745e0d1 100644
--- a/unsloth/models/rl.py
+++ b/unsloth/models/rl.py
@@ -453,5 +453,5 @@ pass
 def PatchFastRL(algorithm = "GRPO", FastLanguageModel = None):
     if FastLanguageModel is not None: PatchRL(FastLanguageModel)
     patch_trl_rl_trainers()
-    PatchRLStatistics(algorithm)
+    if algorithm is nont None: PatchRLStatistics(algorithm)
 pass