diff --git a/face_swapper/src/models/generator.py b/face_swapper/src/models/generator.py
index ec3eadf..a8ae9ff 100644
--- a/face_swapper/src/models/generator.py
+++ b/face_swapper/src/models/generator.py
@@ -2,7 +2,7 @@ import configparser
 
 from torch import Tensor, nn
 
-from ..networks.aienet import AIENet
+from ..networks.aad import AAD
 from ..networks.unet import UNet, UNetPro
 from ..types import Attributes, Embedding
 
@@ -22,7 +22,7 @@ class Generator(nn.Module):
 			self.encoder = UNet()
 		if encoder_type == 'unet-pro':
 			self.encoder = UNetPro()
-		self.generator = AIENet(identity_channels, output_channels, num_blocks)
+		self.generator = AAD(identity_channels, output_channels, num_blocks)
 		self.encoder.apply(init_weight)
 		self.generator.apply(init_weight)
 
diff --git a/face_swapper/src/networks/aienet.py b/face_swapper/src/networks/aad.py
similarity index 99%
rename from face_swapper/src/networks/aienet.py
rename to face_swapper/src/networks/aad.py
index 1b2bee3..de8852c 100644
--- a/face_swapper/src/networks/aienet.py
+++ b/face_swapper/src/networks/aad.py
@@ -4,7 +4,7 @@ from torch import Tensor, nn
 from ..types import Attributes, Embedding
 
 
-class AIENet(nn.Module):
+class AAD(nn.Module):
 	def __init__(self, identity_channels : int, output_channels : int, num_blocks : int) -> None:
 		super().__init__()
 		self.pixel_shuffle_up_sample = PixelShuffleUpSample(identity_channels, output_channels)