From ecc5b1a1d5da2a397364959f0f0a2b4abc2f39d4 Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Tue, 11 Mar 2025 19:43:24 +0100
Subject: [PATCH] Fix test

---
 face_swapper/tests/test_networks.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/face_swapper/tests/test_networks.py b/face_swapper/tests/test_networks.py
index 20c645e..37a9c1a 100644
--- a/face_swapper/tests/test_networks.py
+++ b/face_swapper/tests/test_networks.py
@@ -15,7 +15,7 @@ def test_aad_with_unet(output_size : int) -> None:
 	{
 		'training.model.generator':
 		{
-			'identity_channels': '512',
+			'source_channels': '512',
 			'output_channels': str(output_size * 16),
 			'output_size': str(output_size),
 			'num_blocks': '2'