Fixed TODO higlighting