Remove TODO comment