diff --git a/src/mattext/models/predict.py b/src/mattext/models/predict.py
index 5642df3..490d1d4 100644
--- a/src/mattext/models/predict.py
+++ b/src/mattext/models/predict.py
@@ -82,4 +82,4 @@ def predict(self):
         prediction_ids = self.tokenized_test_datasets["mbid"]
         self.prediction_ids = preidction_ids
 
-        return pd.Series(predictions.predictions.flatten()),preidction_ids
+        return pd.Series(predictions.predictions.flatten()),prediction_ids