diff --git a/unsloth/models/rl_replacements.py b/unsloth/models/rl_replacements.py
index dd4d5a0e8f..06ae82140b 100644
--- a/unsloth/models/rl_replacements.py
+++ b/unsloth/models/rl_replacements.py
@@ -164,11 +164,11 @@ RL_FUNCTIONS["grpo_trainer"].append(grpo_trainer__prepare_inputs)
 # Remove _move_model_to_vllm
 def grpo_trainer__move_model_to_vllm(function_name, function):
     if  function_name != "_move_model_to_vllm": return function
+    
+    def _move_model_to_vllm(self, *args, **kwargs): return None
 
-    print(function)
-    # .*? matches first match. .+? matches final match.
-    replacement = "def _move_model_to_vllm(self, *args, **kwargs): return None\n"
-    return " "*function.find("def") + replacement
+    function = inspect.getsource(_move_model_to_vllm)
+    return function
 pass
 RL_FUNCTIONS["grpo_trainer"].append(grpo_trainer__move_model_to_vllm)