diff --git a/app/models/shakespeare.py b/app/models/shakespeare.py
index f611bf6..b4db11b 100644
--- a/app/models/shakespeare.py
+++ b/app/models/shakespeare.py
@@ -10,11 +10,8 @@ from sqlalchemy import (
     select,
 )
 from sqlalchemy.ext.asyncio import AsyncSession
-from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, relationship
-
-
-class Base(DeclarativeBase):
-    pass
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+from app.models.base import Base
 
 
 class Character(Base):