diff --git a/pkg/api/grpc/echo.go b/pkg/api/grpc/echo.go
index 3313586..2e85094 100644
--- a/pkg/api/grpc/echo.go
+++ b/pkg/api/grpc/echo.go
@@ -14,7 +14,7 @@ type echoServer struct {
 }
 
 func (s *echoServer) Echo (ctx context.Context, message *echo.Message) (*echo.Message, error){
-	// Log level 0 for Info
-	s.logger.Log(0,"Received message body from client:", zap.String("input body", message.Body))
+
+	s.logger.Info("Received message body from client:", zap.String("input body", message.Body))
 	return &echo.Message {Body: message.Body}, nil
 }
diff --git a/pkg/api/grpc/echo_test.go b/pkg/api/grpc/echo_test.go
index e7a526b..4f1812d 100644
--- a/pkg/api/grpc/echo_test.go
+++ b/pkg/api/grpc/echo_test.go
@@ -22,12 +22,13 @@ func TestGrpcEcho(t *testing.T) {
 		lis.Close()
 	})
 
+	s := NewMockGrpcServer()
 	srv := grpc.NewServer()
 	t.Cleanup(func() {
 		srv.Stop()
 	})
 
-	echo.RegisterEchoServiceServer(srv, &echoServer{})
+	echo.RegisterEchoServiceServer(srv, &echoServer{config: s.config, logger: s.logger})
 
 	go func(){
 		if err := srv.Serve(lis); err != nil {