Fix tree_display javascript