Fix async integration test