server/options.go

@@ -147,8 +147,11 @@ func Wait(wg *sync.WaitGroup) options.Option {
 	if wg == nil {
 		wg = new(sync.WaitGroup)
 	}
+
+	wrap := msync.WrapWaitGroup(wg)
+
 	return func(src interface{}) error {
-		return options.Set(src, wg, ".Wait")
+		return options.Set(src, wrap, ".Wait")
 	}
 }