diff --git a/unsloth/models/rl_replacements.py b/unsloth/models/rl_replacements.py
index 9405fef577..0063ea4af6 100644
--- a/unsloth/models/rl_replacements.py
+++ b/unsloth/models/rl_replacements.py
@@ -142,7 +142,7 @@ def grpo_trainer__move_model_to_vllm(function_name, function):
     if  function_name != "_move_model_to_vllm": return function
 
     # .*? matches first match. .+? matches final match.
-    function = "def _move_model_to_vllm(*args, **kwargs): return None\n"
+    function = "def _move_model_to_vllm(self, *args, **kwargs): return None\n"
     return function.find("def") * " " + function
 pass
 RL_FUNCTIONS["grpo_trainer"].append(grpo_trainer__move_model_to_vllm)