fix one test after the API changes