[IMPROVEMENT] fix comment editing