diff --git a/configs/ddrnet_segmentation_model.yaml b/configs/ddrnet_segmentation_model.yaml
index 25779f4a..2bd3b7e8 100644
--- a/configs/ddrnet_segmentation_model.yaml
+++ b/configs/ddrnet_segmentation_model.yaml
@@ -6,7 +6,7 @@ model:
   predefined_model:
     name: DDRNetSegmentationModel
     params:
-      task: multiclass
+      task: binary
       backbone_params:
         use_aux_heads: True # set to False to disable auxiliary heads (for export)
         variant: '23-slim'