diff --git a/src/pilgrim/database.py b/src/pilgrim/database.py
index a7b2a28..9eefcd0 100644
--- a/src/pilgrim/database.py
+++ b/src/pilgrim/database.py
@@ -1,5 +1,5 @@
 from sqlalchemy import create_engine
-from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import declarative_base
 from sqlalchemy.orm import sessionmaker
 from pathlib import Path
 import os