Fixed unit test to match changes in test data.