diff --git a/unsloth/kernels/cross_entropy_loss.py b/unsloth/kernels/cross_entropy_loss.py
index 0034448be5..1fec5d7a85 100644
--- a/unsloth/kernels/cross_entropy_loss.py
+++ b/unsloth/kernels/cross_entropy_loss.py
@@ -443,9 +443,8 @@ 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
+del function, replacement, inspect, re
 
 
 def patch_llama_for_causal_lm():