diff --git a/unsloth/kernels/cross_entropy_loss.py b/unsloth/kernels/cross_entropy_loss.py
index 59d0f3b694..58104e4dfa 100644
--- a/unsloth/kernels/cross_entropy_loss.py
+++ b/unsloth/kernels/cross_entropy_loss.py
@@ -440,6 +440,10 @@ def patch_llama_for_causal_lm():
         function[ function.find("    if not return_dict"):]
     function = function.replace("logits = logits.float()", "\n")
 
+    function = function.split("\n")
+    function = [" "*4 + x for x in function]
+    function = "\n".join(function)
+    
     patched_function = f"class Unsloth_LlamaForCausalLM(LlamaForCausalLM):\n"\
     f"    {function}\n"