Fix expected output