Revert code back to what it was... This fixes the "problem" with highlights.