diff --git a/lib/generators/dockerfile_generator.rb b/lib/generators/dockerfile_generator.rb
index 1250d1d..0adb403 100644
--- a/lib/generators/dockerfile_generator.rb
+++ b/lib/generators/dockerfile_generator.rb
@@ -1599,7 +1599,7 @@ def fly_make_toml
   def fix_database_config
     yaml = IO.read("config/database.yml")
 
-    production = YAML.load(yaml, aliases: true)["production"]
+    production = YAML.load(yaml, aliases: true)["production"] rescue nil
     return unless production.is_a?(Hash) && production.values.all?(Hash)
     return if production.keys == [ "primary" ]