diff --git a/lib/generators/solid_cache/install/templates/config/cache.yml.tt b/lib/generators/solid_cache/install/templates/config/cache.yml.tt
index 70a2ac6..49df92b 100644
--- a/lib/generators/solid_cache/install/templates/config/cache.yml.tt
+++ b/lib/generators/solid_cache/install/templates/config/cache.yml.tt
@@ -1,4 +1,5 @@
 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 %>
@@ -12,5 +13,4 @@ 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 045e90f..37034cb 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,6 +30,7 @@ class SolidCache::InstallGeneratorTest < Rails::Generators::TestCase
       def expected_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 %>
@@ -43,7 +44,6 @@ def expected_cache_config
             <<: *default
 
           production:
-            database: cache
             <<: *default
         YAML
       end