Revert "correct TODO"

This reverts commit 5ab32c8d36388f58fd12f42eb0c9143eb138e268.
fix TODO