Fix tree diagram