diff --git a/face_swapper/src/networks/unet.py b/face_swapper/src/networks/unet.py
index c102b18..c12f3da 100644
--- a/face_swapper/src/networks/unet.py
+++ b/face_swapper/src/networks/unet.py
@@ -4,7 +4,7 @@ from typing import Tuple
 import torch
 from torch import Tensor, nn
 
-from face_swapper.src.types import Feature
+from ..types import Feature
 
 
 class UNet(nn.Module):