diff --git a/src/llamafactory/data/processors/supervised.py b/src/llamafactory/data/processors/supervised.py
index 3981fa12..f488dfca 100644
--- a/src/llamafactory/data/processors/supervised.py
+++ b/src/llamafactory/data/processors/supervised.py
@@ -148,7 +148,7 @@ def preprocess_packed_supervised_dataset(
             logger.warning("Dropped invalid example: {}".format(examples["prompt"][i] + examples["response"][i]))
             continue
 
-        input_ids, labels = _encode_supervised_example(
+        input_ids, labels, _ = _encode_supervised_example(
             prompt=examples["prompt"][i],
             response=examples["response"][i],
             system=examples["system"][i],