Corrects expected output