Fix the highlight