Change TODO comment