Correct highlight when searching