diff --git a/src/mednet/libs/segmentation/engine/predictor.py b/src/mednet/libs/segmentation/engine/predictor.py
index ecac93592137f5bf4e560e2f36d969c5b368bd58..a33a3b80cd786ebf17a9220bd513a7597a649655 100644
--- a/src/mednet/libs/segmentation/engine/predictor.py
+++ b/src/mednet/libs/segmentation/engine/predictor.py
@@ -134,11 +134,11 @@ def run(
     model
         Neural network model (e.g. pasa).
     datamodule
-        The lightning DataModule to use for training **and** validation.
+        The lightning DataModule to run predictions on.
     device_manager
-        An internal device representation, to be used for training and
-        validation.  This representation can be converted into a pytorch device
-        or a lightning accelerator setup.
+        An internal device representation, to be used for prediction. This
+        representation can be converted into a pytorch device or a lightning
+        accelerator setup.
     output_folder
         Folder where to store HDF5 representations of probability maps.