diff --git a/lib/database_rewinder/cleaner.rb b/lib/database_rewinder/cleaner.rb
index 144c131..93f30b3 100644
--- a/lib/database_rewinder/cleaner.rb
+++ b/lib/database_rewinder/cleaner.rb
@@ -14,7 +14,7 @@ def initialize(config: nil, connection_name: nil, only: nil, except: nil)
     end
 
     def db
-      config['database']
+      config.fetch('database') { config[:database] }
     end
 
     def clean(multiple: true)