Fixed the broken async tests