lib/lutaml/model/xml_adapter/builder/nokogiri.rb in lutaml-model-0.3.11 vs lib/lutaml/model/xml_adapter/builder/nokogiri.rb in lutaml-model-0.3.12

@@ -61,14 +61,12 @@
             xml[prefix] if prefix
 
             self
           end
 
-          def method_missing(method_name, *args)
+          def method_missing(method_name, *args, &block)
             if block_given?
-              xml.public_send(method_name, *args) do
-                yield(xml)
-              end
+              xml.public_send(method_name, *args, &block)
             else
               xml.public_send(method_name, *args)
             end
           end