diff --git a/unsloth/kernels/cross_entropy_loss.py b/unsloth/kernels/cross_entropy_loss.py
index 1174ead7d1..49bdbd10ea 100644
--- a/unsloth/kernels/cross_entropy_loss.py
+++ b/unsloth/kernels/cross_entropy_loss.py
@@ -443,6 +443,7 @@ function = [function[0]] + [" "*4 + x for x in function[1:]]
 function = "\n".join(function)
 function = f"class Unsloth_LlamaForCausalLM(LlamaForCausalLM):\n"\
 f"    {function}\n"
+print(function)
 exec(function, globals())
 del function, replacement