modify the comments

you've fixed the bug but without modifing the comment.