spec/riak/map_reduce_spec.rb in riak-client-1.0.0.beta vs spec/riak/map_reduce_spec.rb in riak-client-1.0.0

@@ -2,10 +2,10 @@
 
 describe Riak::MapReduce do
   before :each do
     @client = Riak::Client.new
     @backend = mock("Backend")
-    @client.stub!(:backend).and_return(@backend)
+    @client.stub!(:backend).and_yield(@backend)
     @mr = Riak::MapReduce.new(@client)
   end
 
   it "should require a client" do
     lambda { Riak::MapReduce.new }.should raise_error