fixed tree diagram to work with new VisBuilder implementation