Tweak generate TODOs