diff --git a/face_swapper/src/models/discriminator.py b/face_swapper/src/models/discriminator.py
index 63da839..5a2865c 100644
--- a/face_swapper/src/models/discriminator.py
+++ b/face_swapper/src/models/discriminator.py
@@ -44,7 +44,7 @@ class MultiscaleDiscriminator(nn.Module):
 class NLayerDiscriminator(nn.Module):
 	def __init__(self, input_channels : int, num_filters : int, num_layers : int, kernel_size : int) -> None:
 		super(NLayerDiscriminator, self).__init__()
-		layers = self.create_layers(input_channels, num_filters, num_layers, kernel_size)
+		layers = self.create_layers(self, input_channels, num_filters, num_layers, kernel_size)
 		self.discriminator = nn.Sequential(*layers)
 
 	@staticmethod