diff --git a/StreamServer/src/database.py b/StreamServer/src/database.py
index 20712da..7802a3c 100644
--- a/StreamServer/src/database.py
+++ b/StreamServer/src/database.py
@@ -11,8 +11,7 @@ Attributes:
 """
 
 from sqlalchemy import create_engine
-from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.orm import sessionmaker
+from sqlalchemy.orm import sessionmaker, declarative_base
 
 from minio import Minio
 from config import DB_HOST, DB_USER, DB_PASSWD, DB_NAME, MINIO_ENDPOINT, MINIO_ACCESS_KEY, MINIO_SECRET_KEY