diff --git a/src/backend/base/langflow/api/build.py b/src/backend/base/langflow/api/build.py
index ef05da8b4..7fb32e5c3 100644
--- a/src/backend/base/langflow/api/build.py
+++ b/src/backend/base/langflow/api/build.py
@@ -424,7 +424,9 @@ async def generate_flow_events(
         )
         event_manager.on_error(data=error_message.data)
         raise
+
     event_manager.on_end(data={})
+    await graph.end_all_traces()
     await event_manager.queue.put((None, None, time.time()))