fix highlighter issue in edit mode