diff --git a/__init__.py b/__init__.py
index 7f3881a..6f4e249 100644
--- a/__init__.py
+++ b/__init__.py
@@ -6,6 +6,7 @@ from .src.nodes import (
     ComfyUIClipDTO,
     ComfyUIConditioningDTO,
     ComfyUIPipelineDTO,
+    ComfyUISchedulerDTO,
 )
 
 load_envs()
@@ -17,4 +18,5 @@ __all__ = [
     "ComfyUIAutoencoderDTO",
     "ComfyUIClipDTO",
     "ComfyUIConditioningDTO",
+    "ComfyUISchedulerDTO",
 ]