diff --git a/src/fastapi_app/lib/models/base_sqlalchemy.py b/src/fastapi_app/lib/models/base_sqlalchemy.py
index e3af184..b0d8b5d 100644
--- a/src/fastapi_app/lib/models/base_sqlalchemy.py
+++ b/src/fastapi_app/lib/models/base_sqlalchemy.py
@@ -1,3 +1,5 @@
-import sqlalchemy.ext.declarative
+import sqlalchemy.orm
 
-Base = sqlalchemy.ext.declarative.declarative_base()
+
+class Base(sqlalchemy.orm.DeclarativeBase):
+    pass