1 files changed, 5 insertions, 2 deletions
diff --git a/examples/asyncio/greenlet_orm.py b/examples/asyncio/greenlet_orm.py
index 7429b6853..92880b992 100644
--- a/examples/asyncio/greenlet_orm.py
+++ b/examples/asyncio/greenlet_orm.py
@@ -10,13 +10,16 @@ from sqlalchemy import Column
 from sqlalchemy import ForeignKey
 from sqlalchemy import Integer
 from sqlalchemy import String
+from sqlalchemy.ext.asyncio import AsyncAttrs
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy.ext.asyncio import create_async_engine
-from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.future import select
+from sqlalchemy.orm import DeclarativeBase
 from sqlalchemy.orm import relationship
 
-Base = declarative_base()
+
+class Base(AsyncAttrs, DeclarativeBase):
+    pass
 
 
 class A(Base):