fixed doctest errors, resulting from change to Tree api