diff --git a/Client/Python/Calc_Client.py b/Client/Python/Calc_Client.py
index e8439bf..d296306 100644
--- a/Client/Python/Calc_Client.py
+++ b/Client/Python/Calc_Client.py
@@ -3,10 +3,11 @@
 import grpc
 
 import Calc_pb2
+import Calc_pb2_grpc
 
 def run():
   channel = grpc.insecure_channel('localhost:50050')
-  stub = Calc_pb2.CalculatorStub(channel)
+  stub = Calc_pb2_grpc.CalculatorStub(channel)
   response = stub.Add(Calc_pb2.AddRequest(n1=20,n2=10))
   print(response.n1)
   response = stub.Substract(Calc_pb2.SubstractRequest(n1=20,n2=10))
diff --git a/Server/Calc_Server.py b/Server/Calc_Server.py
index 2cb9c91..9e45b52 100644
--- a/Server/Calc_Server.py
+++ b/Server/Calc_Server.py
@@ -4,11 +4,12 @@
 import grpc
 
 import Calc_pb2
+import Calc_pb2_grpc
 
 _ONE_DAY_IN_SECONDS = 60 * 60 * 24
 
 
-class Calculator(Calc_pb2.CalculatorServicer):
+class Calculator(Calc_pb2_grpc.CalculatorServicer):
 
   def Add(self, request, context):
     return Calc_pb2.AddReply(n1=request.n1+request.n2)
@@ -25,7 +26,7 @@ def Divide(self, request, context):
 
 def serve():
   server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
-  Calc_pb2.add_CalculatorServicer_to_server(Calculator(), server)
+  Calc_pb2_grpc.add_CalculatorServicer_to_server(Calculator(), server)
   server.add_insecure_port('[::]:50050')
   server.start()
   try:
@@ -35,4 +36,4 @@ def serve():
     server.stop(0)
 
 if __name__ == '__main__':
-  serve()
\ No newline at end of file
+  serve()