Add TODO.

TODO: remove duplicate code.