lib/rspec/mocks/instance_method_stasher.rb in rspec-mocks-3.4.0 vs lib/rspec/mocks/instance_method_stasher.rb in rspec-mocks-3.4.1

@@ -52,9 +52,10 @@
 
         # @private
         def stash
           return unless method_defined_directly_on_klass?
           @original_method ||= ::RSpec::Support.method_handle_for(@object, @method)
+          @klass.__send__(:undef_method, @method)
         end
 
         # @private
         def restore
           return unless @original_method