diff --git a/models/performance_metric.py b/models/performance_metric.py
index 9d22098..0304fef 100644
--- a/models/performance_metric.py
+++ b/models/performance_metric.py
@@ -3,9 +3,8 @@ from __future__ import annotations
 from datetime import datetime
 
 from sqlalchemy import Column, DateTime, Float, Integer, String
-from sqlalchemy.ext.declarative import declarative_base
 
-Base = declarative_base()
+from config.database import Base
 
 
 class PerformanceMetric(Base):