small TODO update