Fix a TODO comment.