fixed edge case on highlight function