diff --git a/core/cpu_deformer.py b/core/cpu_deformer.py
index cbf140f..6895a3f 100644
--- a/core/cpu_deformer.py
+++ b/core/cpu_deformer.py
@@ -1,6 +1,6 @@
 import numpy as np
 from PIL import Image
-from core.base_landmarks import MediapipeBaseLandmarks
+from ..core.base_landmarks import MediapipeBaseLandmarks
 
 class CPUDeformer:
     """
diff --git a/core/gpu_deformer.py b/core/gpu_deformer.py
index b7e38ff..53f73d9 100644
--- a/core/gpu_deformer.py
+++ b/core/gpu_deformer.py
@@ -1,7 +1,7 @@
 import numpy as np
 from PIL import Image
 import torch
-from core.base_landmarks import MediapipeBaseLandmarks
+from ..core.base_landmarks import MediapipeBaseLandmarks
 
 
 class GPUDeformer: