diff --git a/{{cookiecutter.project_slug}}/backend/app/app/db/base_class.py b/{{cookiecutter.project_slug}}/backend/app/app/db/base_class.py
index 2228a37..3dadcb3 100644
--- a/{{cookiecutter.project_slug}}/backend/app/app/db/base_class.py
+++ b/{{cookiecutter.project_slug}}/backend/app/app/db/base_class.py
@@ -1,11 +1,9 @@
-from sqlalchemy.ext.declarative import declarative_base, declared_attr
+from sqlalchemy.ext.declarative import as_declarative, declared_attr
 
 
-class CustomBase(object):
+@as_declarative()
+class Base:
     # Generate __tablename__ automatically
     @declared_attr
     def __tablename__(cls):
         return cls.__name__.lower()
-
-
-Base = declarative_base(cls=CustomBase)