Fixed leftover TODOs