restore wg.Add and wg.Done