Updated TODO. Fixed comment. [ci skip]