diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index ea6647594d..0c9468ce9d 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -1166,7 +1166,7 @@ def _unsloth_pre_compute_loss(self, model, inputs, *args, **kwargs):
             inputs["num_items_in_batch"] = kwargs["num_items_in_batch"]
         pass
     pass
-    return self._old_compute_loss(model, inputs, args, kwargs)
+    return self._old_compute_loss(model, inputs, *args, **kwargs)
 pass