diff --git a/face_swapper/src/networks/masknet.py b/face_swapper/src/networks/masknet.py
index b80655e..0c43c4d 100644
--- a/face_swapper/src/networks/masknet.py
+++ b/face_swapper/src/networks/masknet.py
@@ -3,7 +3,7 @@ from configparser import ConfigParser
 import torch
 from torch import Tensor, nn
 
-from face_swapper.src.types import Attribute
+from ..types import Attribute
 
 
 class MaskNet(nn.Module):