Update tests to the new API.