diff --git a/__init__.py b/__init__.py
index 325cc90..77187b2 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,5 +1,5 @@
-from .gemini_image_node import *
-from .gemini_image_preset_node import *
+from gemini.gemini_image_node import *
+from gemini.gemini_image_preset_node import *
 from typing_extensions import override
 
 class APIExtension(ComfyExtension):
diff --git a/gemini_image_node.py b/gemini/gemini_image_node.py
similarity index 99%
rename from gemini_image_node.py
rename to gemini/gemini_image_node.py
index 65274ab..22e23b8 100644
--- a/gemini_image_node.py
+++ b/gemini/gemini_image_node.py
@@ -9,7 +9,7 @@ from PIL import Image
 from io import BytesIO
 from typing_extensions import override
 from comfy_api.latest import ComfyExtension, io
-from .image_utils import tensor_to_base64_string
+from ..utils.image_utils import tensor_to_base64_string
 
 class GeminiImage(io.ComfyNode):
     """
diff --git a/gemini_image_preset.json b/gemini/gemini_image_preset.json
similarity index 100%
rename from gemini_image_preset.json
rename to gemini/gemini_image_preset.json
diff --git a/gemini_image_preset_node.py b/gemini/gemini_image_preset_node.py
similarity index 100%
rename from gemini_image_preset_node.py
rename to gemini/gemini_image_preset_node.py
diff --git a/image_utils.py b/utils/image_utils.py
similarity index 100%
rename from image_utils.py
rename to utils/image_utils.py