update TODO comment