enable result buffering for postgres

1 files changed, 2 insertions, 2 deletions

diff --git a/update_sharing.py b/update_sharing.py
index 450bfc7..e41bceb 100755
--- a/update_sharing.py
+++ b/update_sharing.py
@@ -4,7 +4,7 @@ import optparse
 
 import sqlalchemy
 
-from dedup.utils import fetchiter, enable_sqlite_foreign_keys
+from dedup.utils import fetchiter, configure_database_engine
 
 def add_values(conn, insert_key, files, size):
     params = dict(files=files, size=size, pid1=insert_key[0],
@@ -78,5 +78,5 @@ if __name__ == "__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)
     main(db)