diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 563a723..c2e3cd2 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -28,4 +28,4 @@ jobs:
      python-version: '3.12'
   - run: pip install torch torchvision
   - run: pip install pytest
-  - run: PYTHONPATH=/Users/runner/work/facefusion-labs/facefusion-labs pytest
+  - run: PYTHONPATH=/home/runner/work/facefusion-labs/facefusion-labs pytest
diff --git a/face_swapper/tests/test_networks.py b/face_swapper/tests/test_networks.py
index f4d9b8b..37ccc7a 100644
--- a/face_swapper/tests/test_networks.py
+++ b/face_swapper/tests/test_networks.py
@@ -1,14 +1,8 @@
-import os
-import sys
-
 import pytest
 import torch
 
-package_directory = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
-sys.path.append(package_directory)
-
-#from face_swapper.src.networks.aad import AAD
-#from face_swapper.src.networks.unet import UNet
+from face_swapper.src.networks.aad import AAD
+from face_swapper.src.networks.unet import UNet
 
 
 @pytest.mark.parametrize('output_size', [ 256 ])
@@ -20,14 +14,13 @@ def test_aad_with_unet(output_size : int) -> None:
 		output_channels = 8192
 	num_blocks = 2
 
-	#generator = AAD(identity_channels, output_channels, output_size, num_blocks).eval()
-	#encoder = UNet(output_size).eval()
+	generator = AAD(identity_channels, output_channels, output_size, num_blocks).eval()
+	encoder = UNet(output_size).eval()
 
 	source_tensor = torch.randn(1, 512)
 	target_tensor = torch.randn(1, 3, output_size, output_size)
 
-	#target_attributes = encoder(target_tensor)
-	#output_tensor = generator(source_tensor, target_attributes)
+	target_attributes = encoder(target_tensor)
+	output_tensor = generator(source_tensor, target_attributes)
 
-	assert package_directory == None
-	#assert output_tensor.shape == (1, 3, output_size, output_size)
+	assert output_tensor.shape == (1, 3, output_size, output_size)