Fix highlight bug.