lib/scaffoldhub/helper.rb in scaffoldhub-0.0.10 vs lib/scaffoldhub/helper.rb in scaffoldhub-0.0.11

@@ -41,11 +41,11 @@
       end
     end
 
     def each_gem
       begin
-        gems = scaffold_spec.gems.each do |gem|
-          yield gem
+        if (gems = scaffold_spec.gems)
+          gems.each { |gem| yield gem }
         end
       rescue Errno::ENOENT => e
         say_status :error, e.message, :red
         nil
       rescue Scaffoldhub::NotFoundException => e