diff --git a/python/paddle/fluid/tests/book/test_recognize_digits.py b/python/paddle/fluid/tests/book/test_recognize_digits.py
index 383f3772253e4e702c85a3725d4cac72bb9dedd6..4a70976a4837c668a5e0ba6d49b598d046a8ec5d 100644
--- a/python/paddle/fluid/tests/book/test_recognize_digits.py
+++ b/python/paddle/fluid/tests/book/test_recognize_digits.py
@@ -146,9 +146,7 @@ def train(nn_type,
                                 params_filename=params_filename)
                         if save_full_dirname is not None:
                             fluid.io.save_inference_model(
-                                save_full_dirname,
-                                None,
-                                None,
+                                save_full_dirname, [], [],
                                 exe,
                                 model_filename=model_filename,
                                 params_filename=params_filename,