Minor TODO update