diff --git a/src/libecalc/presentation/yaml/mappers/component_mapper.py b/src/libecalc/presentation/yaml/mappers/component_mapper.py
index af05783b1a..8a4578e0dc 100644
--- a/src/libecalc/presentation/yaml/mappers/component_mapper.py
+++ b/src/libecalc/presentation/yaml/mappers/component_mapper.py
@@ -228,7 +228,7 @@ def from_yaml_to_dto(
                 user_defined_category=user_defined_category,
                 cable_loss=cable_loss,
                 max_usage_from_shore=max_usage_from_shore,
-                component_type=ComponentType.GENERATOR_SET,  # TODO: Check if this is correct. Why isn´t component_type set for YamlGeneratorSet?
+                component_type=ComponentType.GENERATOR_SET,
             )
         except ValidationError as e:
             raise DtoValidationError(data=data.model_dump(), validation_error=e) from e