update option Wait

2024-03-12 10:59:18 +03:00 · 2024-03-12 10:59:18 +03:00 · 87db21bc18
commit 87db21bc18
parent 43fdf2e83a
1 changed files with 4 additions and 1 deletions
--- a/server/options.go
+++ b/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")
 	}
 }