diff --git a/nce_cost/train.py b/nce_cost/train.py
index 94b761880a93fa5b7d0d42d7128ffb2625fe24b8..3babf7fe0963fcff54430cd174b0af523e68846b 100644
--- a/nce_cost/train.py
+++ b/nce_cost/train.py
@@ -47,8 +47,8 @@ def train(model_save_dir):
             paddle.reader.shuffle(
                 lambda: paddle.dataset.imikolov.train(word_dict, 5)(),
                 buf_size=1000), 64),
-    num_passes=1000,
-    event_handler=event_handler)
+        num_passes=1000,
+        event_handler=event_handler)
 
 
 if __name__ == "__main__":