Fixed crash in tree controller, commented out original one.