lib/yamg/cli.rb in yamg-0.3.0 vs lib/yamg/cli.rb in yamg-0.3.1

@@ -76,9 +76,10 @@
       puts Rainbow("Tasks: #{works.keys.join(', ')}").yellow
       works.each { |out, opts| compile_work(out, opts) }
     end
 
     def screenshot
+      return unless YAMG.config['screenshots'].respond_to?(:each)
       YAMG.config['screenshots'].each do |ss|
         Thread.new do
           Screenshot.new(ss).work('./export')
           puts Rainbow("[o]SS #{ss[0]} #{ss[1]}").black
         end