diff --git a/face_swapper/src/models/generator.py b/face_swapper/src/models/generator.py
index 3ff125f..8953354 100644
--- a/face_swapper/src/models/generator.py
+++ b/face_swapper/src/models/generator.py
@@ -5,7 +5,7 @@ from torch import Tensor, nn
 
 from ..networks.aad import AAD
 from ..networks.unet import UNet
-from ..types import Embedding, Attribute
+from ..types import Attribute, Embedding
 
 
 class Generator(nn.Module):