diff --git a/icedyno/training/dataloaders.py b/icedyno/training/dataloaders.py
index d1b0102..2c7d97d 100644
--- a/icedyno/training/dataloaders.py
+++ b/icedyno/training/dataloaders.py
@@ -69,7 +69,7 @@ class BinaryTargetNetCDFGenerator(tf.keras.utils.Sequence):
     Generator for Keras training to allow multiprocessing and training on batches with only the
     batch itself being loaded into memory.
 
-    Targets a binary target (ice/no ice).
+    Targets a binary image (ice/no ice).
     """
 
     def __init__(