Fixed TODO highlight