Fixing ui-tree call