diff --git a/rattail_tempmon/db/alembic/env.py b/rattail_tempmon/db/alembic/env.py
index b7872ae..5f6d670 100644
--- a/rattail_tempmon/db/alembic/env.py
+++ b/rattail_tempmon/db/alembic/env.py
@@ -6,6 +6,7 @@ Alembic Environment Script
 from __future__ import unicode_literals, absolute_import
 
 from alembic import context
+from sqlalchemy import orm
 
 from rattail.config import make_config
 from rattail.db.config import configure_versioning