Fix tree display