Amend TODO comment