diff --git a/unsloth/kernels/cross_entropy_loss.py b/unsloth/kernels/cross_entropy_loss.py
index 6c1b7bdd0f..f2e77ef4d1 100644
--- a/unsloth/kernels/cross_entropy_loss.py
+++ b/unsloth/kernels/cross_entropy_loss.py
@@ -508,7 +508,7 @@ def cross_entropy_loss(
     ignored_index=-100,
     inplace_backward: bool = False,
     process_group=None,
-) -> Tuple[torch.Tensor, torch.Tensor]:
+) :
     """
     Arguments:
         logits: (batch, vocab_size)