diff --git a/python/paddle_serving_client/__init__.py b/python/paddle_serving_client/__init__.py
index 7a36f370961e755f7c3a2745f5eb4e9e7f80394d..09dd1ae1c7ff8e2f477b0b0dbcae286684c2c513 100644
--- a/python/paddle_serving_client/__init__.py
+++ b/python/paddle_serving_client/__init__.py
@@ -26,7 +26,8 @@ import grpc
 from .proto import multi_lang_general_model_service_pb2
 sys.path.append(
     os.path.join(os.path.abspath(os.path.dirname(__file__)), 'proto'))
-from .proto import grpc_pb2 as grpc_pb2
+from .proto import multi_lang_general_model_service_pb2 as pb2
+from .proto import multi_lang_general_model_service_pb2_grpc as grpc_pb2
 
 int_type = 0
 float_type = 1
@@ -395,7 +396,7 @@ class MultiLangClient(object):
         self._parse_model_config(proto_txt)
 
     def _load_client_config(self, stub):
-        req= grpc_pb2.ServingConfig()
+        req= pb2.ServingConfig()
         self._config  = self.stub_.get_client_proto_text(req)
         self._parse_model_config(config.proto_txt)
 
diff --git a/python/paddle_serving_server/__init__.py b/python/paddle_serving_server/__init__.py
index fbcc138d83bad9d1b3b34bfc0afd189b4af2b7d7..7c8b525ae65d1ac5eef510685d4a44bda9dfa66e 100644
--- a/python/paddle_serving_server/__init__.py
+++ b/python/paddle_serving_server/__init__.py
@@ -441,7 +441,7 @@ class Server(object):
 
 
 class MultiLangServerService(
-        pb2_grpc.MultiLangGeneralModelService):
+        grpc_pb2.MultiLangGeneralModelService):
     def __init__(self, model_config_path, endpoints):
         from paddle_serving_client import Client