Fixing ui tree