fix highlighting bug.