diff --git a/src/database.py b/src/database.py
index de44e86..f51e710 100644
--- a/src/database.py
+++ b/src/database.py
@@ -6,13 +6,14 @@ from flask_sqlalchemy import SQLAlchemy
 from sqlalchemy import Boolean, DateTime, Integer, String
 from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
 
-db: SQLAlchemy = SQLAlchemy()
-
 
 class Base(DeclarativeBase):
     pass
 
 
+db: SQLAlchemy = SQLAlchemy(model_class=Base)
+
+
 class CameraStatus(Base):
     __tablename__ = "camera_status"