diff --git a/data/database.py b/data/database.py
index da18eb30e..de7262287 100644
--- a/data/database.py
+++ b/data/database.py
@@ -306,9 +306,8 @@ def _db_from_url(url, db_kwargs, connect_timeout=DEFAULT_DB_CONNECT_TIMEOUT):
     db_kwargs.pop('stale_timeout', None)
     db_kwargs.pop('max_connections', None)
 
-  # wrapped_driver = _wrap_for_retry(driver)
-  # return wrapped_driver(parsed_url.database, **db_kwargs)
-  return driver(parsed_url.database, **db_kwargs)
+  wrapped_driver = _wrap_for_retry(driver)
+  return wrapped_driver(parsed_url.database, **db_kwargs)
 
 
 def configure(config_object):