Small TODO update