Fix older TODOs