Fixing issue with ui_tree