diff --git a/face_swapper/src/networks/unet.py b/face_swapper/src/networks/unet.py
index a5c3d9a..b70b44d 100644
--- a/face_swapper/src/networks/unet.py
+++ b/face_swapper/src/networks/unet.py
@@ -13,7 +13,7 @@ class UNet(nn.Module):
 		self.up_samples = self.create_up_samples()
 
 	@staticmethod
-	def create_down_samples(self) -> nn.ModuleList:
+	def create_down_samples(self : nn.Module) -> nn.ModuleList:
 		return nn.ModuleList(
 		[
 			nn.Sequential(