diff --git a/lib/spy/strategy/wrap.rb b/lib/spy/strategy/wrap.rb index d2aac36..531f733 100644 --- a/lib/spy/strategy/wrap.rb +++ b/lib/spy/strategy/wrap.rb @@ -8,11 +8,11 @@ def initialize(spy) end def apply - ReplaceMethod.call(@spy.original.owner, @spy, mode: :stub) + ReplaceMethod.call(@spy.original.owner, @spy, mode: :stub, remove_existing: true) end def undo - ReplaceMethod.call(@spy.original.owner, @spy, mode: :restore) + ReplaceMethod.call(@spy.original.owner, @spy, mode: :restore, remove_existing: true) end end end