diff --git a/__init__.py b/__init__.py
index 1160962..60e71a3 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,4 +1,4 @@
-from video_extract_key_frames.nodes import ExtractKeyFrames, ExtractKeyFramesAndSplit
+from .video_extract_key_frames.nodes import ExtractKeyFrames, ExtractKeyFramesAndSplit
 
 # A dictionary that contains all nodes you want to export with their names
 # NOTE: names should be globally unique
diff --git a/test/__init__.py b/test/__init__.py
index fc6f5cb..bc16892 100644
--- a/test/__init__.py
+++ b/test/__init__.py
@@ -1,16 +1,16 @@
-"""Top-level package for extractstoryboards."""
-
-__all__ = [
-    "NODE_CLASS_MAPPINGS",
-    "NODE_DISPLAY_NAME_MAPPINGS",
-    "WEB_DIRECTORY",
-]
-
-__author__ = """guyouyue"""
-__email__ = "guyuoyue@gmail.com"
-__version__ = "1.0.0"
-
-from .src.extractstoryboards.nodes import NODE_CLASS_MAPPINGS
-from .src.extractstoryboards.nodes import NODE_DISPLAY_NAME_MAPPINGS
-
-WEB_DIRECTORY = "./web"
+# """Top-level package for extractstoryboards."""
+#
+# __all__ = [
+#     "NODE_CLASS_MAPPINGS",
+#     "NODE_DISPLAY_NAME_MAPPINGS",
+#     "WEB_DIRECTORY",
+# ]
+#
+# __author__ = """guyouyue"""
+# __email__ = "guyuoyue@gmail.com"
+# __version__ = "1.0.0"
+#
+# from .src.extractstoryboards.nodes import NODE_CLASS_MAPPINGS
+# from .src.extractstoryboards.nodes import NODE_DISPLAY_NAME_MAPPINGS
+#
+# WEB_DIRECTORY = "./web"
diff --git a/video_extract_key_frames/__init__.py b/video_extract_key_frames/__init__.py
index e69de29..aa09235 100644
--- a/video_extract_key_frames/__init__.py
+++ b/video_extract_key_frames/__init__.py
@@ -0,0 +1,9 @@
+from .nodes import ExtractKeyFrames, ExtractKeyFramesAndSplit
+from .service import extract_keyframes, split_and_save_video_segments
+
+__all__ = [
+    'ExtractKeyFrames',
+    'ExtractKeyFramesAndSplit',
+    'extract_keyframes',
+    'split_and_save_video_segments'
+]
\ No newline at end of file
diff --git a/video_extract_key_frames/nodes.py b/video_extract_key_frames/nodes.py
index 410fc39..39fccac 100644
--- a/video_extract_key_frames/nodes.py
+++ b/video_extract_key_frames/nodes.py
@@ -2,7 +2,7 @@
 # 主要是各个节点的定义和业务逻辑
 import folder_paths
 
-from video_extract_key_frames.service import (
+from .service import (
     extract_keyframes,
     split_and_save_video_segments
 )