diff --git a/src/dispatch/function.py b/src/dispatch/function.py index 71de6500..798af504 100644 --- a/src/dispatch/function.py +++ b/src/dispatch/function.py @@ -36,8 +36,10 @@ def decorator(f): def method(self, *args, **kwargs): if len(args) == 1 and len(kwargs) == 0 and callable(args[0]): return f(self, args[0]) + def wrapper(func): return f(self, func, *args, **kwargs) + return wrapper return method