diff --git a/api/core/entities/provider_configuration.py b/api/core/entities/provider_configuration.py
index 26659e1056..6cf9a403e3 100644
--- a/api/core/entities/provider_configuration.py
+++ b/api/core/entities/provider_configuration.py
@@ -979,7 +979,7 @@ class ProviderConfiguration(BaseModel):
                     lb_config.enabled = False
                     lb_config.name = "__delete__"
                     lb_config.updated_at = datetime.datetime.now(datetime.UTC).replace(tzinfo=None)
-                    db.session.add(lb_config)
+                    session.add(lb_config)
 
                 # Check if this is the currently active credential
                 provider_model_record = self._get_custom_model_record(model_type, model, session=session)