diff --git a/python/paddle/distributed/parallel.py b/python/paddle/distributed/parallel.py
index d9d252024d9f3b9439c83591c93523d01c81e750..f0365cab8c896689ba2f49020878e3236d3381a9 100644
--- a/python/paddle/distributed/parallel.py
+++ b/python/paddle/distributed/parallel.py
@@ -30,6 +30,7 @@ from paddle.fluid.dygraph import parallel_helper
 from paddle.distributed.fleet.launch_utils import check_backend
 from paddle.fluid.dygraph.parallel import ParallelEnv
 from paddle.distributed.fleet.base.private_helper_function import wait_server_ready  # noqa: F401
+from paddle.distributed import collective
 from paddle.distributed.collective import _set_group_map
 from paddle.distributed.collective import _set_group_map_by_name
 from paddle.distributed.collective import _get_group_map_by_name
diff --git a/python/paddle/fluid/contrib/slim/tests/save_quant_model.py b/python/paddle/fluid/contrib/slim/tests/save_quant_model.py
index f97c2778c0918ecbfbed546089c17e9d505818cd..73ec8cf3e023d0a4c3bc7844b1582cc80e45103a 100644
--- a/python/paddle/fluid/contrib/slim/tests/save_quant_model.py
+++ b/python/paddle/fluid/contrib/slim/tests/save_quant_model.py
@@ -88,8 +88,8 @@ def transform_and_save_int8_model(original_path,
                                   debug=False,
                                   quant_model_filename='',
                                   quant_params_filename='',
-                                  save_model_filename='',
-                                  save_params_filename=''):
+                                  save_model_filename="__model__",
+                                  save_params_filename=None):
     place = fluid.CPUPlace()
     exe = fluid.Executor(place)
     inference_scope = fluid.executor.global_scope()