Remove hacks/unnecessary TODOs