diff --git a/api/core/rag/retrieval/dataset_retrieval.py b/api/core/rag/retrieval/dataset_retrieval.py
index 11010c9d60..1c1019e18c 100644
--- a/api/core/rag/retrieval/dataset_retrieval.py
+++ b/api/core/rag/retrieval/dataset_retrieval.py
@@ -534,7 +534,6 @@ class DatasetRetrieval:
                                     synchronize_session=False,
                                 )
                             )
-                            db.session.commit()
                     else:
                         query = db.session.query(DocumentSegment).where(
                             DocumentSegment.index_node_id == document.metadata["doc_id"]
@@ -596,7 +595,7 @@ class DatasetRetrieval:
         with flask_app.app_context():
             with Session(db.engine) as session:
                 dataset_stmt = select(Dataset).where(Dataset.id == dataset_id)
-                dataset = db.session.scalar(dataset_stmt)
+                dataset = session.scalar(dataset_stmt)
 
             if not dataset:
                 return []