change the comment in code so that it correct