diff --git a/api/core/callback_handler/index_tool_callback_handler.py b/api/core/callback_handler/index_tool_callback_handler.py
index c663408559..64c734f626 100644
--- a/api/core/callback_handler/index_tool_callback_handler.py
+++ b/api/core/callback_handler/index_tool_callback_handler.py
@@ -53,9 +53,9 @@ class DatasetIndexToolCallbackHandler:
                         ChildChunk.document_id == dataset_document.id,
                     ).first()
                     if child_chunk:
-                        segment = DocumentSegment.query.filter(
-                            DocumentSegment.id == child_chunk.segment_id
-                        ).update({DocumentSegment.hit_count: DocumentSegment.hit_count + 1}, synchronize_session=False)
+                        segment = DocumentSegment.query.filter(DocumentSegment.id == child_chunk.segment_id).update(
+                            {DocumentSegment.hit_count: DocumentSegment.hit_count + 1}, synchronize_session=False
+                        )
                 else:
                     query = db.session.query(DocumentSegment).filter(
                         DocumentSegment.index_node_id == document.metadata["doc_id"]
diff --git a/api/core/rag/retrieval/dataset_retrieval.py b/api/core/rag/retrieval/dataset_retrieval.py
index 173622aa01..c5ac63e853 100644
--- a/api/core/rag/retrieval/dataset_retrieval.py
+++ b/api/core/rag/retrieval/dataset_retrieval.py
@@ -440,9 +440,9 @@ class DatasetRetrieval:
                         ChildChunk.document_id == dataset_document.id,
                     ).first()
                     if child_chunk:
-                        segment = DocumentSegment.query.filter(
-                            DocumentSegment.id == child_chunk.segment_id
-                        ).update({DocumentSegment.hit_count: DocumentSegment.hit_count + 1}, synchronize_session=False)
+                        segment = DocumentSegment.query.filter(DocumentSegment.id == child_chunk.segment_id).update(
+                            {DocumentSegment.hit_count: DocumentSegment.hit_count + 1}, synchronize_session=False
+                        )
                         db.session.commit()
                 else:
                     query = db.session.query(DocumentSegment).filter(