improve TODO comment