diff --git a/plugins/module_utils/mysql.py b/plugins/module_utils/mysql.py
index af026f3..cc60f68 100644
--- a/plugins/module_utils/mysql.py
+++ b/plugins/module_utils/mysql.py
@@ -145,14 +145,14 @@ def mysql_connect(module, login_user=None, login_password=None, config_file='',
         # In case of PyMySQL driver:
         if mysql_driver.version_info[0] < 1:
             # for PyMySQL < 1.0.0, use 'db' instead of 'database'
-            config['db'] = config['database']
+            config['db'] = config.get('database')
             del config['database']
         db_connection = mysql_driver.connect(autocommit=autocommit, **config)
     else:
         # In case of MySQLdb driver
         if mysql_driver.version_info[0] < 2 or mysql_driver.version_info[1] < 1:
             # for MySQLdb < 2.1.0, use 'db' instead of 'database'
-            config['db'] = config['database']
+            config['db'] = config.get('database')
             del config['database']
         db_connection = mysql_driver.connect(**config)
         if autocommit: