diff --git a/tools/export_serving_model.py b/tools/export_serving_model.py
index 112acd5524494845f847b46629924b81e192a42e..07c9e7d77eaadf2b5fd58bd791533f0b2003bf34 100644
--- a/tools/export_serving_model.py
+++ b/tools/export_serving_model.py
@@ -50,8 +50,12 @@ def save_serving_model(FLAGS, exe, feed_vars, test_fetches, infer_prog):
     import paddle_serving_client.io as serving_io
     serving_client = os.path.join(save_dir, 'serving_client')
     serving_server = os.path.join(save_dir, 'serving_server')
-    serving_io.save_model(serving_client, serving_server, feed_dict, fetch_dict,
-                          infer_prog)
+    serving_io.save_model(
+        client_config_folder=serving_client,
+        server_model_folder=serving_server,
+        feed_dict,
+        fetch_dict,
+        infer_prog)
 
 
 def main():