fix: return the decorated function, not the decorator