diff --git a/torchgeo/datamodules/loveda.py b/torchgeo/datamodules/loveda.py
index b1fd8d78c6c..a82233f9844 100644
--- a/torchgeo/datamodules/loveda.py
+++ b/torchgeo/datamodules/loveda.py
@@ -77,6 +77,7 @@ def setup(self, stage: Optional[str] = None) -> None:
             split="val", transforms=val_predict_transforms, **self.kwargs
         )
 
+        # Test set masks are not public, use for prediction instead
         self.predict_dataset = LoveDA(
             split="test", transforms=val_predict_transforms, **self.kwargs
         )