Fixed bug on load_tree_from_pickle.