diff --git a/treelib/tree.py b/treelib/tree.py index c2594f3..2635a8e 100644 --- a/treelib/tree.py +++ b/treelib/tree.py @@ -1976,7 +1976,9 @@ def to_dict( ) if len(tree_dict[ntag]["children"]) == 0: tree_dict = self[nid].tag if not with_data else {ntag: {"data": self[nid].data}} - return tree_dict + else: + tree_dict = self[nid].tag if not with_data else {ntag: {"data": self[nid].data}} + return tree_dict def to_json(self, with_data: bool = False, sort: bool = True, reverse: bool = False): """