Update sub_tree_test.cpp