diff --git a/face_swapper/tests/test_networks.py b/face_swapper/tests/test_networks.py
index 37ccc7a..dbc1773 100644
--- a/face_swapper/tests/test_networks.py
+++ b/face_swapper/tests/test_networks.py
@@ -1,6 +1,10 @@
+import sys
+
 import pytest
 import torch
 
+sys.path.append('..')
+
 from face_swapper.src.networks.aad import AAD
 from face_swapper.src.networks.unet import UNet