diff --git a/ee/cli/plugins/sitedb.py b/ee/cli/plugins/sitedb.py
index 805b72bc..cb7e1bde 100644
--- a/ee/cli/plugins/sitedb.py
+++ b/ee/cli/plugins/sitedb.py
@@ -4,7 +4,7 @@ from sqlalchemy.orm import relationship, backref
 from sqlalchemy.ext.declarative import declarative_base
 from ee.core.logging import Log
 from ee.core.database import db_session
-from ee.core.models import SiteDB
+from ee.cli.plugins.models import SiteDB
 import sys