corrected return to tree from test