Minor update in the comments