Fixed TODOs