Remove TODO comment that's been completed