diff --git a/decor/on_complete.go b/decor/on_complete.go index bb4a563..714a0de 100644 --- a/decor/on_complete.go +++ b/decor/on_complete.go @@ -10,6 +10,10 @@ d := &onCompleteWrapper{ Decorator: decorator, msg: message, + } + if md, ok := decorator.(*mergeDecorator); ok { + d.Decorator, md.Decorator = md.Decorator, d + return md } return d }