removed TODO comment
minor change