Fix highlightig