Fix TOC highlight