diff --git a/api/models/base.py b/api/models/base.py
index da9509301a..0c2963b0e4 100644
--- a/api/models/base.py
+++ b/api/models/base.py
@@ -1,5 +1,6 @@
 from sqlalchemy.orm import declarative_base
 
-from models.engine import metadata
+from models.engine import db, metadata
 
 Base = declarative_base(metadata=metadata)
+Base.query = db.session.query_property()