diff --git a/ms2pip/_utils/ion_mobility.py b/ms2pip/_utils/ion_mobility.py
index 8a3764b..7dceb9f 100644
--- a/ms2pip/_utils/ion_mobility.py
+++ b/ms2pip/_utils/ion_mobility.py
@@ -9,7 +9,7 @@
 
 
 class IonMobility:
-    """Predict ion mobility using IM²Deep."""
+    """Predict ion mobility using IM2Deep."""
 
     def __init__(self, processes=1) -> None:
         # Lazy import to avoid loading loading heavy dependencies when not needed
@@ -27,6 +27,6 @@ def add_im_predictions(self, psm_list: PSMList) -> None:
         """Add ion mobility predictions to the PSMList."""
         logger.info("Predicting ion mobility...")
         predictions: pd.Series = self.predict_fn(
-            psm_list, write_output=False, n_jobs=self.processes
+            psm_list, write_output=False, n_jobs=self.processes, ion_mobility=True
         )
         psm_list["ion_mobility"] = predictions.values