diff --git a/api/core/provider_manager.py b/api/core/provider_manager.py
index df4157c1f2..da5fcaf458 100644
--- a/api/core/provider_manager.py
+++ b/api/core/provider_manager.py
@@ -3,7 +3,7 @@ from collections import defaultdict
 from json import JSONDecodeError
 from typing import Optional
 
-from psycopg2 import IntegrityError
+from sqlalchemy.exc import IntegrityError
 
 from core.entities.model_entities import DefaultModelEntity, DefaultModelProviderEntity
 from core.entities.provider_configuration import ProviderConfigurations, ProviderConfiguration, ProviderModelBundle