diff --git a/application.py b/application.py
index 63c1e2d8a..0b119f5d2 100644
--- a/application.py
+++ b/application.py
@@ -1,8 +1,8 @@
 import logging
 
 from app import app as application
+from data.model import db as model_db
 
-import model
 
 logging.basicConfig(**application.config['LOGGING_CONFIG'])
 
@@ -30,7 +30,7 @@ application.register_blueprint(webhooks, url_prefix='/webhooks')
 
 
 def close_db(exc):
-  db = model.db
+  db = model_db
   if not db.is_closed():
     logger.debug('Disconnecting from database.')
     db.close()