diff --git a/pysegcnn/core/dataset.py b/pysegcnn/core/dataset.py
index a3371c0f222c6547699c50817cbd6d3d054dd522..ee36ba3953957f197d03cb1bb10048acff7e260c 100644
--- a/pysegcnn/core/dataset.py
+++ b/pysegcnn/core/dataset.py
@@ -668,10 +668,10 @@ class Cloud95Dataset(ImageDataset):
 
 
 class SupportedDatasets(enum.Enum):
-    Sparcs = {'name': 'Sparcs', 'class': SparcsDataset}
-    Cloud95 = {'name': 'Cloud95', 'class': Cloud95Dataset}
-    Garmisch = {'name': 'Garmisch', 'class': ProSnowGarmisch}
-    Obergurgl = {'name': 'Obergurgl', 'class': ProSnowObergurgl}
+    Sparcs = SparcsDataset
+    Cloud95 = Cloud95Dataset
+    Garmisch = ProSnowGarmisch
+    Obergurgl = ProSnowObergurgl
 
 
 if __name__ == '__main__':