diff --git a/dooit/ui/widgets/trees/_decorators.py b/dooit/ui/widgets/trees/_decorators.py index d5a9b3e2..66b6c6a3 100644 --- a/dooit/ui/widgets/trees/_decorators.py +++ b/dooit/ui/widgets/trees/_decorators.py @@ -29,8 +29,9 @@ def wrapper(self: "ModelTree", *args, **kwargs) -> Any: def refresh_tree(func: Callable) -> Callable: def wrapper(self: "ModelTree", *args, **kwargs) -> Any: - func(self, *args, **kwargs) + res = func(self, *args, **kwargs) self.force_refresh() + return res return wrapper