diff --git a/src/nodes/_diffusers_scheduler_loader.py b/src/nodes/_diffusers_scheduler_loader.py
index 9795352..d518d78 100644
--- a/src/nodes/_diffusers_scheduler_loader.py
+++ b/src/nodes/_diffusers_scheduler_loader.py
@@ -3,7 +3,7 @@ from dependency_injector.wiring import Provide, inject
 from ..di import Container
 from ..ui import SchedulerHandler
 from .dto import ComfyUIPipelineDTO, ComfyUISchedulerDTO
-from .dto.type import ComfyUISchedulerType
+from .type import ComfyUISchedulerType
 
 
 class DiffusersSchedulerLoader:
diff --git a/src/nodes/dto/type/__init__.py b/src/nodes/type/__init__.py
similarity index 100%
rename from src/nodes/dto/type/__init__.py
rename to src/nodes/type/__init__.py
diff --git a/src/nodes/dto/type/_scheduler_type.py b/src/nodes/type/_scheduler_type.py
similarity index 98%
rename from src/nodes/dto/type/_scheduler_type.py
rename to src/nodes/type/_scheduler_type.py
index 2bd8586..3ba1dbc 100644
--- a/src/nodes/dto/type/_scheduler_type.py
+++ b/src/nodes/type/_scheduler_type.py
@@ -1,4 +1,4 @@
-from ....domain.model import Scheduler
+from ...domain.model import Scheduler
 
 
 class ComfyUISchedulerType: