lib/yasf/scraper.rb in yasf-0.0.5 vs lib/yasf/scraper.rb in yasf-0.0.6

@@ -117,10 +117,10 @@
     def initialize(source, options = nil)
       @options = options || {}
       case source
       when String
         @document = Nokogiri::HTML(self.class.get(source,@options).body)
-      when Nokogiri::XML::Element
+      when Nokogiri::XML::Element, Nokogiri::HTML::Document
         @document = source
       else
         raise ArgumentError, "source not recognized"
       end
     end