diff --git a/api/db/bookmark_db.py b/api/db/bookmark_db.py
index 42fe451..0bb98af 100644
--- a/api/db/bookmark_db.py
+++ b/api/db/bookmark_db.py
@@ -9,6 +9,7 @@
 import datetime
 import sqlite3
 import uuid
+from .db import Database
 
 class BookmarkDatabase(Database):
     """
diff --git a/api/db/telethon_db.py b/api/db/telethon_db.py
index 343d0d3..1bb0cf5 100644
--- a/api/db/telethon_db.py
+++ b/api/db/telethon_db.py
@@ -10,6 +10,7 @@
 import datetime
 import sqlite3
 import uuid
+from .db import Database
 
 class TelethonDatabase(Database):
     """