lib/abstract_mapper/builder.rb in abstract_mapper-0.0.2 vs lib/abstract_mapper/builder.rb in abstract_mapper-0.1.0

@@ -37,11 +37,11 @@
     # @param [Proc] block
     #   The block with DSL commands for adding subnodes
     #
     # @return (see #tree)
     #
-    def self.update(node = Branch.new, &block)
+    def self.update(node = AST::Branch.new, &block)
       new(node, &block).tree
     end
 
     # @!attribute [r] tree
     #
@@ -70,10 +70,10 @@
 
     private # DSL commands
 
     def method_missing(name, *args, &block)
       node  = @commands[name].call(*args, &block)
-      @tree = tree << (node.is_a?(Branch) ? update(node, &block) : node)
+      @tree = tree << (node.is_a?(AST::Branch) ? update(node, &block) : node)
     end
 
     def respond_to_missing?(*)
       true
     end