Adjust expected output