diff --git a/db/python/tables/cohort.py b/db/python/tables/cohort.py
index 94ef9bc95..e16809227 100644
--- a/db/python/tables/cohort.py
+++ b/db/python/tables/cohort.py
@@ -2,7 +2,7 @@
 import dataclasses
 import datetime
 
-from db.python.connect import DbBase
+from db.python.tables.base import DbBase
 from db.python.tables.project import ProjectId
 from db.python.utils import GenericFilter, GenericFilterModel
 from models.models.cohort import Cohort