Fixed the build, added the missing import from IterationTreeView