diff --git a/main.py b/main.py
index c8ee51b..94cd41f 100644
--- a/main.py
+++ b/main.py
@@ -155,7 +155,7 @@
                 num_inputs, num_hidden, mask, num_cond_inputs,
                 s_act='tanh', t_act='relu')
         ]
-    mask = 1 - mask
+        mask = 1 - mask
 elif args.flow == 'realnvp':
     mask = torch.arange(0, num_inputs) % 2
     mask = mask.to(device).float()