diff --git a/rattail/db/__init__.py b/rattail/db/__init__.py
index f46ee2f27e091450204f33ffef6bada55c43ebde..034d3e56ee304c327c93d829ab25de68469b1cca 100644
--- a/rattail/db/__init__.py
+++ b/rattail/db/__init__.py
@@ -37,9 +37,11 @@ def init(config):
     Initialize the Rattail database framework.
     """
 
-    from rattail.db import extension
-    edbob.graft(rattail, extension.model)
-    edbob.graft(rattail, extension.enum)
+    from rattail.db.extension import model
+    edbob.graft(rattail, model)
+
+    from rattail.db.extension import enum
+    edbob.graft(rattail, enum)
 
 
 def init_database(engine, session):