diff --git a/py/modules/human_parsing/parsing_api.py b/py/modules/human_parsing/parsing_api.py
index c0df8f1..610d1c3 100644
--- a/py/modules/human_parsing/parsing_api.py
+++ b/py/modules/human_parsing/parsing_api.py
@@ -4,7 +4,7 @@ import cv2
 import torchvision.transforms as transforms
 from torch.utils.data import DataLoader
 from .simple_extractor_dataset import SimpleFolderDataset
-from transforms import transform_logits
+from .transforms import transform_logits
 from tqdm import tqdm
 from PIL import Image
 
diff --git a/py/nodes/image.py b/py/nodes/image.py
index 41c910d..01922db 100644
--- a/py/nodes/image.py
+++ b/py/nodes/image.py
@@ -1348,7 +1348,7 @@ class humanSegmentation:
         if method in cache:
           _, parsing = cache[method][1]
         else:
-          from ..modules.human_parsing import HumanParsing
+          from ..modules.human_parsing.run_parsing import HumanParsing
           onnx_path = os.path.join(folder_paths.models_dir, 'onnx')
           model_path = get_local_filepath(HUMANPARSING_MODELS['parsing_lip']['model_url'], onnx_path)
           parsing = HumanParsing(model_path=model_path)