diff --git a/lib/generators/solid_cache/install/templates/config/solid_cache.yml.tt b/lib/generators/solid_cache/install/templates/config/solid_cache.yml.tt
index 49df92b..70a2ac6 100644
--- a/lib/generators/solid_cache/install/templates/config/solid_cache.yml.tt
+++ b/lib/generators/solid_cache/install/templates/config/solid_cache.yml.tt
@@ -1,5 +1,4 @@
 default: &default
-  database: <%= ENV.fetch("DATABASE", "cache") %>
   store_options:
     # Cap age of oldest cache entry to fulfill retention policies
     # max_age: <%%= 60.days.to_i %>
@@ -13,4 +12,5 @@ test:
   <<: *default
 
 production:
+  database: <%= ENV.fetch("DATABASE", "cache") %>
   <<: *default
diff --git a/test/lib/generators/solid_cache/solid_cache/install_generator_test.rb b/test/lib/generators/solid_cache/solid_cache/install_generator_test.rb
index 3942788..9b01d9d 100644
--- a/test/lib/generators/solid_cache/solid_cache/install_generator_test.rb
+++ b/test/lib/generators/solid_cache/solid_cache/install_generator_test.rb
@@ -30,7 +30,6 @@ class SolidCache::InstallGeneratorTest < Rails::Generators::TestCase
       def expected_solid_cache_config
         <<~YAML
           default: &default
-            database: cache
             store_options:
               # Cap age of oldest cache entry to fulfill retention policies
               # max_age: <%= 60.days.to_i %>
@@ -44,6 +43,7 @@ def expected_solid_cache_config
             <<: *default
 
           production:
+            database: cache
             <<: *default
         YAML
       end