Fix for make_tree script