diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index d22407d..b6ced3e 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -31,4 +31,4 @@ jobs:
      python-version: '3.12'
   - run: pip install torch torchvision
   - run: pip install pytest
-  - run: pytest
+  - run: pytest face_swapper/tests
diff --git a/face_swapper/tests/test_networks.py b/face_swapper/tests/test_networks.py
index e1d4381..18ce42d 100644
--- a/face_swapper/tests/test_networks.py
+++ b/face_swapper/tests/test_networks.py
@@ -1,8 +1,8 @@
 import torch
 import pytest
 
-from face_swapper.src.networks.aad import AAD
-from face_swapper.src.networks.unet import UNet
+from ..src.networks.aad import AAD
+from ..src.networks.unet import UNet
 
 
 @pytest.mark.parametrize('output_size', [ 256 ])