Update the tests to match the API changes.