diff --git a/rattail/db.py b/rattail/db.py
index 49eb0d5bb39a9642484b2b1f9d2fff106f4124ef..4f8a5e93115f5298399c3e09d7246c2f1d41326a 100644
--- a/rattail/db.py
+++ b/rattail/db.py
@@ -31,12 +31,12 @@ from edbob.db.extensions import activate_extension
 import rattail
 
 
-def init_database(session):
+def init_database(engine, session):
     """
     Initialize an ``edbob`` database for use with Rattail.
     """
 
-    activate_extension('rattail', session.bind)
+    activate_extension('rattail', engine)
 
     columns = [
         ('F01',         'UPC',                  'GPC(14)'),