(Nearly) Fix all tests with async magic.