diff --git a/api/core/callback_handler/index_tool_callback_handler.py b/api/core/callback_handler/index_tool_callback_handler.py
index f0c9379413..669e0829a5 100644
--- a/api/core/callback_handler/index_tool_callback_handler.py
+++ b/api/core/callback_handler/index_tool_callback_handler.py
@@ -31,8 +31,17 @@ class DatasetIndexToolCallbackHandler(IndexToolCallbackHandler):
             index_node_id = node.node.doc_id
 
             # add hit count to document segment
-            db.session.query(DocumentSegment).filter(
-                DocumentSegment.dataset_id == self.dataset_id,
-                DocumentSegment.index_node_id == index_node_id
-            ).update({DocumentSegment.hit_count: DocumentSegment.hit_count + 1}, synchronize_session=False)
+            update_stmt = (
+                db.session.query(DocumentSegment)
+                .filter(
+                    DocumentSegment.dataset_id == self.dataset_id,
+                    DocumentSegment.index_node_id == index_node_id
+                )
+                .update(
+                    {DocumentSegment.hit_count: DocumentSegment.hit_count + 1},
+                    synchronize_session=False
+                )
+            )
 
+            db.session.execute(update_stmt)
+            db.session.commit()