Updated test to reflect changes in the API.