update the TODO comment per review feedback