Updated TODO comment.