diff --git a/train_ms.py b/train_ms.py
index 7299116..4e6fad9 100644
--- a/train_ms.py
+++ b/train_ms.py
@@ -140,7 +140,7 @@ def run(rank, n_gpus, hps):
         hps.train.learning_rate,
         betas=hps.train.betas,
         eps=hps.train.eps)
-   if net_dur_disc is not None:
+    if net_dur_disc is not None:
         optim_dur_disc = torch.optim.AdanW(
         net_dur_disc.parameters(),
         hps.train.learning_rate,