diff --git a/unsloth/kernels/cross_entropy_loss.py b/unsloth/kernels/cross_entropy_loss.py
index a139fce49d..2010a4cbe4 100644
--- a/unsloth/kernels/cross_entropy_loss.py
+++ b/unsloth/kernels/cross_entropy_loss.py
@@ -217,7 +217,6 @@ class Fast_CrossEntropyLoss(torch.autograd.Function):
                 losses,
                 logsumexp,
                 labels,
-                n_rows,
                 VOCAB_SIZE = vocab_size,
                 BLOCK_SIZE = MAX_FUSED_SIZE,
                 num_warps  = 32,