lib/sugar-high/module.rb in sugar-high-0.1.0 vs lib/sugar-high/module.rb in sugar-high-0.1.1

@@ -1,21 +1,21 @@
 require 'active_support/inflector'
 
 def modules *module_names, &block
-  module_names.each do |name|
+  module_names.flatten.each do |name|
     class_eval %{
       module #{name.to_s.camelize}
         #{yield block if block}
       end
     }    
   end
 end 
 
 def nested_modules *module_names, &block
-  module_names.inject([]) do |res, name|  
+  module_names.flatten.inject([]) do |res, name|  
     res << %{
       module #{name.to_s.camelize}
         #{yield block if block}
       end}  
-  end.join("\n")
+  end.flatten.join("\n")
 end