Fix lost TODOs