diff --git a/grpc.go b/grpc.go
index 972f4ba..689cd7d 100644
--- a/grpc.go
+++ b/grpc.go
@@ -118,6 +118,8 @@ func (g *grpcServer) configure(opts ...server.Option) {
 		o(&g.opts)
 	}
 
+	g.wg = wait(g.opts.Context)
+
 	maxMsgSize := g.getMaxMsgSize()
 
 	gopts := []grpc.ServerOption{