diff --git a/train_ms.py b/train_ms.py
index 38d1687..1bda2a5 100644
--- a/train_ms.py
+++ b/train_ms.py
@@ -454,9 +454,7 @@ def train_and_evaluate(
 
                 loss_fm = feature_loss(fmap_r, fmap_g)
                 loss_gen, losses_gen = generator_loss(y_d_hat_g)
-                loss_gen_all = (
-                    loss_gen + loss_fm + loss_mel + loss_dur + loss_kl
-                )
+                loss_gen_all = loss_gen + loss_fm + loss_mel + loss_dur + loss_kl
                 if net_dur_disc is not None:
                     loss_dur_gen, losses_dur_gen = generator_loss(y_dur_hat_g)
                     loss_gen_all += loss_dur_gen