Remove TODO from comment