lib/better_html/ast/node.rb in better_html-1.0.1 vs lib/better_html/ast/node.rb in better_html-1.0.2

@@ -4,11 +4,15 @@
 module BetterHtml
   module AST
     class Node < ::AST::Node
       attr_reader :loc
 
-      def descendants(type, &block)
-        AST::Iterator.descendants(self, type, &block)
+      def descendants(*types)
+        AST::Iterator.descendants(self, types)
+      end
+
+      def location
+        loc
       end
     end
   end
 end