Minor. Fixes TODOs