Fixed the test to match the new data