diff --git a/service/backend/app/model/wand_model.py b/service/backend/app/model/wand_model.py
index ebebb84..0175513 100644
--- a/service/backend/app/model/wand_model.py
+++ b/service/backend/app/model/wand_model.py
@@ -19,7 +19,7 @@
 from pydantic import BaseModel
 from sqlalchemy import create_engine, Column, Text, TIMESTAMP, func, Integer, BigInteger
 from sqlalchemy.dialects.postgresql import JSONB
-from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import declarative_base
 from sqlalchemy.schema import UniqueConstraint, Index
 
 from ..wand_env import REDIS_POOL, SERVER_URL