Updated TODO after merging #6.

[ci skip]