modify the comments

you've fixed the bug but without modifing the comment. 
this line is added by elloop