Improving Async Test with sync.WaitGroup