Corrects expected output from code