lib/architecture-js/generator.rb in architecture-js-0.3.1 vs lib/architecture-js/generator.rb in architecture-js-0.3.2

@@ -9,18 +9,22 @@
     def initialize(blueprint)
       @project = blueprint
       @blueprint = @project.config
       @template_paths = @project.template_directories
       @templates = Hash.new
-      find_templates @template_paths
+      load_templates
     end
 
     def find_templates(paths)
-      paths.each do |path|
+      [*paths].each do |path|
         Dir["#{path}/*"].each do |file|
           add_file_to_templates file
         end
       end
+    end
+
+    def load_templates
+      find_templates @template_paths
     end
 
     def add_file_to_templates(file)
       ext = File.extname(file)
       template_name = File.basename(file, ext)
\ No newline at end of file