fixing TODO comment