Fixed the output test case