Tiny mods to TODO