Fix tree output