Enhance the test case to make the output simpler