[minor] Fix comments