Fix async tests