diff --git a/config/dataset/test_ci_segmentation_binary_RGBL.yaml b/config/dataset/test_ci_segmentation_binary_RGBL.yaml
index 7e261262..d9851954 100644
--- a/config/dataset/test_ci_segmentation_binary_RGBL.yaml
+++ b/config/dataset/test_ci_segmentation_binary_RGBL.yaml
@@ -14,6 +14,6 @@ dataset:
   attribute_field: properties/class
   attribute_values: [1]
   class_name: # will follow in the next version
-  classes_dict: {'WAER':1, 'FORE':2, 'ROAI':3, 'BUIL':4}
+  classes_dict: {'BUIL':1}
   class_weights:
   ignore_index: 255
diff --git a/config/dataset/test_ci_segmentation_multiclass_RGBL.yaml b/config/dataset/test_ci_segmentation_multiclass_RGBL.yaml
new file mode 100644
index 00000000..7e261262
--- /dev/null
+++ b/config/dataset/test_ci_segmentation_multiclass_RGBL.yaml
@@ -0,0 +1,19 @@
+# @package _global_
+dataset:
+  # dataset-wide
+  name:
+  raw_data_csv: ${general.raw_data_csv}
+  raw_data_dir: ${general.raw_data_dir}
+  download_data: False
+
+  # imagery
+  bands: [R,G,B]
+  dsm_dir: null
+
+  # ground truth
+  attribute_field: properties/class
+  attribute_values: [1]
+  class_name: # will follow in the next version
+  classes_dict: {'WAER':1, 'FORE':2, 'ROAI':3, 'BUIL':4}
+  class_weights:
+  ignore_index: 255
diff --git a/config/gdl_config_template.yaml b/config/gdl_config_template.yaml
index 0e07e720..29d8089a 100644
--- a/config/gdl_config_template.yaml
+++ b/config/gdl_config_template.yaml
@@ -7,7 +7,7 @@ defaults:
   - optimizer: adamw
   - callbacks: default_callbacks
   - scheduler: plateau
-  - dataset: test_ci_segmentation_binary_RGBL
+  - dataset: test_ci_segmentation_binary
   - augmentation: basic_augmentation_segmentation
   - tracker: # set logger here or use command line (e.g. `python GDL.py tracker=mlflow`)
   - visualization: default_visualization