enable result buffering for postgres

1 files changed, 2 insertions, 2 deletions

diff --git a/readyaml.py b/readyaml.py
index 15cfcb3..8ba82fa 100755
--- a/readyaml.py
+++ b/readyaml.py
@@ -9,7 +9,7 @@ from debian.debian_support import version_compare
 import sqlalchemy
 import yaml
 
-from dedup.utils import enable_sqlite_foreign_keys
+from dedup.utils import configure_database_engine
 
 def readyaml(conn, stream):
     gen = yaml.safe_load_all(stream)
@@ -64,7 +64,7 @@ def main():
                       help="location of the database")
     options, args = parser.parse_args()
     db = sqlalchemy.create_engine(options.database)
-    enable_sqlite_foreign_keys(db)
+    configure_database_engine(db)
     with db.begin() as conn:
         readyaml(conn, sys.stdin)