fix a copy paste error and update TODO