Update tests and test data for API changes