diff --git a/face_swapper/src/models/loss.py b/face_swapper/src/models/loss.py
index ccbd96a..1cca867 100644
--- a/face_swapper/src/models/loss.py
+++ b/face_swapper/src/models/loss.py
@@ -64,8 +64,8 @@ class CycleLoss(nn.Module):
 			temp_tensors.append(temp_tensor)
 
 		feature_loss = torch.stack(temp_tensors).mean()
-		mae_loss = self.mae_loss(target_tensor, cycle_tensor)
-		cycle_loss = (feature_loss + mae_loss) * 0.5
+		reconstruction_loss = self.mae_loss(target_tensor, cycle_tensor)
+		cycle_loss = (feature_loss + reconstruction_loss) * 0.5
 		weighted_feature_loss = cycle_loss * self.config_cycle_weight
 		return cycle_loss, weighted_feature_loss