fix the highlighter bug