diff --git a/unsloth/models/rl_replacements.py b/unsloth/models/rl_replacements.py
index 7d46ea21b6..55c2daa323 100644
--- a/unsloth/models/rl_replacements.py
+++ b/unsloth/models/rl_replacements.py
@@ -90,6 +90,8 @@ def sft_trainer_prepare_dataset(function_name, function):
         if matched:
             # Use fast version!
             function = inspect.getsource(fast_sft_prepare_dataset)
+            function = function.split("\n")
+            function = "\n".join(" "*4 + x for x in function)
             function = function.replace("def sft_prepare_dataset", "def _prepare_dataset")
             return function
         pass