lib/tako/query_chain.rb in tako-0.4.0 vs lib/tako/query_chain.rb in tako-0.4.1

@@ -6,15 +6,13 @@
     def initialize(proxy, base_object)
       @proxy = proxy
       @base_object = base_object
     end
 
-    def method_missing(method, *args)
+    def method_missing(method, *args, &block)
       @proxy.with_shard do
         result = if block_given?
-                    base_object.send(method, *args) do
-                      yield
-                    end
+                    base_object.send(method, *args, &block)
                   else
                     base_object.send(method, *args)
                   end
 
         if chain_available?(result)