Fix tree toggle