Async unit test fix