diff --git a/easyapi/ImageNode.py b/easyapi/ImageNode.py
index 73e7fed..8f732c5 100644
--- a/easyapi/ImageNode.py
+++ b/easyapi/ImageNode.py
@@ -8,7 +8,7 @@ from comfy.cli_args import args
 from PIL.PngImagePlugin import PngInfo
 import json
 from json import JSONEncoder, JSONDecoder
-from util import tensor_to_pil
+from easyapi.util import tensor_to_pil
 
 class Base64ToImage:
     @classmethod
diff --git a/easyapi/SamNode.py b/easyapi/SamNode.py
index 57ff93f..da6ee60 100644
--- a/easyapi/SamNode.py
+++ b/easyapi/SamNode.py
@@ -2,7 +2,7 @@ from segment_anything import SamAutomaticMaskGenerator
 import json
 import numpy as np
 
-from util import tensor_to_pil
+from easyapi.util import tensor_to_pil
 
 class SamAutoMaskSEGS:
     @classmethod