diff --git a/src/terratorch/tasks/segmentation_tasks.py b/src/terratorch/tasks/segmentation_tasks.py
index 6e7fe8cf..7070b9c5 100644
--- a/src/terratorch/tasks/segmentation_tasks.py
+++ b/src/terratorch/tasks/segmentation_tasks.py
@@ -158,7 +158,7 @@ def configure_losses(self) -> None:
         elif loss == "dice":
             self.criterion = smp.losses.DiceLoss("multiclass", ignore_index=ignore_index)
         else:
-            exception_message = f"Loss type '{loss}' is not valid. Currently, supports 'ce', 'jaccard' or 'focal' loss."
+            exception_message = f"Loss type '{loss}' is not valid. Currently, supports 'ce', 'jaccard', 'dice' or 'focal' loss."
             raise ValueError(exception_message)
 
     def configure_metrics(self) -> None: