add missing imports in model-tree