diff --git a/python/paddle/fluid/tests/unittests/test_checkpoint.py b/python/paddle/fluid/tests/unittests/test_checkpoint.py
index cf70dfd448363c85a257a1c32c68ad67343a15b6..e22400a045ced16c46b0bf005155f621f249d263 100644
--- a/python/paddle/fluid/tests/unittests/test_checkpoint.py
+++ b/python/paddle/fluid/tests/unittests/test_checkpoint.py
@@ -66,9 +66,9 @@ class TestCheckpoint(unittest.TestCase):
 
             exe = fluid.Executor(self.place)
             for i in xrange(10):
-                fluid.io.save_checkpoint(
-                    exe, config.checkpoint_dir, self.trainer_id, self.chief,
-                    trainer_args, program, config.max_num_checkpoints)
+                fluid.io.save_checkpoint(exe, config.checkpoint_dir,
+                                         self.trainer_id, trainer_args, program,
+                                         config.max_num_checkpoints)
 
 
 if __name__ == '__main__':