fix in highlighting