spec/support/actor_mocking.rb in ridley-1.0.0.rc1 vs spec/support/actor_mocking.rb in ridley-1.0.0.rc2

@@ -1,9 +1,9 @@
 RSpec.configuration.before(:each) do
   class Celluloid::ActorProxy
     unless @rspec_compatible
       @rspec_compatible = true
-      undef_method :should_receive
-      undef_method :stub
+      undef_method :should_receive if method_defined?(:should_receive)
+      undef_method :stub if method_defined?(:stub)
     end
   end
 end