diff --git a/core/lib/generators/refinery/cms/cms_generator.rb b/core/lib/generators/refinery/cms/cms_generator.rb
index <HASH>..<HASH> 100644
--- a/core/lib/generators/refinery/cms/cms_generator.rb
+++ b/core/lib/generators/refinery/cms/cms_generator.rb
@@ -39,7 +39,7 @@ module Refinery
   # Refinery has set config.assets.initialize_on_precompile = false by default.
   config.assets.initialize_on_precompile = false
 
-},                       :after => "Application.configure do\n" if env == 'production'
+},                       :after => "Application.configure do\n" if env =~ /production/
       end
 
       # Stop pretending
Use regular expression to detect if current env contains 'production'.