Fix minor issue with test case output