diff --git a/unsloth/kernels/cross_entropy_loss.py b/unsloth/kernels/cross_entropy_loss.py
index 4b4bda0b4e..184b08c674 100644
--- a/unsloth/kernels/cross_entropy_loss.py
+++ b/unsloth/kernels/cross_entropy_loss.py
@@ -106,7 +106,7 @@ def _large_cross_entropy_forward(
         if  (label_idx >=    (col_idx+0)*BLOCK_SIZE) and \
             (label_idx < min((col_idx+1)*BLOCK_SIZE, n_cols)):
 
-            loss = tl.load(logits_ptr + label_idx).to(tl.float32)
+            logits_label = tl.load(logits_ptr + label_idx).to(tl.float32)
             lse  = 0.0
             loss = lse - logits_label # We add the final logsumexp after a reduction
         pass