fixed output to match test cases