fix TaskIterator import statement