Clean up TODO and typo.