Fixed the test to match the output