[minor] Fixing comments