Updated TODO comment