Fixed TreeItem bug.

Removed old TreeItem import.