diff --git a/app/db.py b/app/db.py
index 0e993f1..ae48155 100644
--- a/app/db.py
+++ b/app/db.py
@@ -4,13 +4,12 @@ from sqlalchemy.orm import DeclarativeBase, MappedAsDataclass
 
 db = SQLAlchemy()
 
-
-class Base(DeclarativeBase, MappedAsDataclass):
-    pass
-
-
-class ImagesCache(Base):
+class ImagesCache(db.Model):
     __tablename__ = "image_cache"
 
     codename: Mapped[str] = mapped_column(primary_key=True)
     image: Mapped[str] = mapped_column()
+
+    def __init__(self, codename, image):
+        self.codename = codename
+        self.image = image