Fix broken test not waiting on Async method to complete