Fixed a TODO.