Remove TODO
Update comment.