fix of TODO comment