fix comment refactor