Updated test to match service changes.