fix comment in function