diff --git a/setup.py b/setup.py
index 68c824c..fdb6ada 100644
--- a/setup.py
+++ b/setup.py
@@ -1,8 +1,7 @@
 import os
 
-import sqlalchemy.engine.url
 from dotenv import load_dotenv
-from sqlalchemy import create_engine
+from sqlalchemy import URL, create_engine
 from sqlalchemy.orm import sessionmaker
 
 load_dotenv()
@@ -11,7 +10,7 @@ testing: bool = False
 
 db_password = os.getenv("DB_PASSWORD")
 
-db_uri = sqlalchemy.engine.url.URL(
+db_uri = URL.create(
     drivername="postgresql+psycopg2",
     username="staw_db",
     password=db_password,