fixes for TODOs