diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index 5024204f20..acf411b6c9 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -1192,7 +1192,7 @@ for j, function in enumerate(functions):
         try: exec(f"EMPTY_LOGITS.{function} = warn_logits_error",  globals(), locals())
         except: continue
     else:
-        exec(f"def raise_{j}: print({function})", globals(), locals())
+        exec(f"def raise_{j}(*args, **kwargs): print({function})", globals(), locals())
         try: exec(f"EMPTY_LOGITS.{function} = raise_{j}", globals(), locals())
         except: continue
 pass