Fix TODO highlighting