Fixed expected results to match the test file