Correct comment in code