diff --git a/vescale/utils/monkey_patch.py b/vescale/utils/monkey_patch.py index b6cc622..71d8513 100644 --- a/vescale/utils/monkey_patch.py +++ b/vescale/utils/monkey_patch.py @@ -20,7 +20,7 @@ def patch_method(target: Any, method_name: str, raise_error=False): def helper(func): - fn = getattr(target, "method_name", None) + fn = getattr(target, method_name, None) if fn is None: setattr(target, method_name, func)