fix treedraw doctest