Correct expected output of a few test cases