Fix error highlighting disappearing