diff --git a/nodes.py b/nodes.py
index fb941f9..1c22381 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1,8 +1,8 @@
 
 import folder_paths
 import torch
-from seecoder import SemanticExtractionEncoder, QueryTransformer, Decoder
-from swin import SwinTransformer
+from .seecoder import SemanticExtractionEncoder, QueryTransformer, Decoder
+from .swin import SwinTransformer
 import safetensors.torch
 import os
 import sys