Fix minor prob on get_height_method_tree