Fixes TODOs