fixed TODO comment