Fix for async testing