diff --git a/server/models.py b/server/models.py
index dc6e4b7..6adf283 100644
--- a/server/models.py
+++ b/server/models.py
@@ -35,10 +35,11 @@ from sqlalchemy import (
     Text,
     func,
 )
-from sqlalchemy.orm import declarative_base, relationship
+from sqlalchemy.orm import DeclarativeBase, relationship
 
 
-Base = declarative_base()
+class Base(DeclarativeBase):
+    __allow_unmapped__ = True
 
 
 class User(Base):
@@ -270,4 +271,4 @@ class TaskRun(Base):
     status: str = Column(String(50), nullable=False)
     message: Optional[str] = Column(Text, nullable=True)
 
-    task: BackupTask = relationship("BackupTask", back_populates="runs")
\ No newline at end of file
+    task: BackupTask = relationship("BackupTask", back_populates="runs")