[minor] Fixed the comments