diff --git a/lib/generators/sequel.rb b/lib/generators/sequel.rb
index 6c13e1f..6b8e8ad 100644
--- a/lib/generators/sequel.rb
+++ b/lib/generators/sequel.rb
@@ -11,7 +11,8 @@ def self.base_name
         @_sequel_base_name ||= 'sequel'
       end
 
-      def self.source_root
+      def self.source_root(path = nil)
+        @_sequel_source_root = path if path
         @_sequel_source_root ||= File.expand_path(
           "../#{base_name}/#{generator_name}/templates", __FILE__
         )