diff --git a/web/app/components/workflow-app/hooks/__tests__/use-workflow-run.spec.ts b/web/app/components/workflow-app/hooks/__tests__/use-workflow-run.spec.ts
index 86a85bf70d..8f9911f58b 100644
--- a/web/app/components/workflow-app/hooks/__tests__/use-workflow-run.spec.ts
+++ b/web/app/components/workflow-app/hooks/__tests__/use-workflow-run.spec.ts
@@ -442,10 +442,12 @@ describe('useWorkflowRun', () => {
       workflow_id: 'flow-1',
       reason: 'failed',
       node_type: 'llm',
-      workflow_status: WorkflowRunningStatus.Running,
-      workflow_tracing_count: 1,
-      workflow_data: workflowData,
-      workflow_data_json: JSON.stringify(workflowData),
+      data: {
+        workflow_status: WorkflowRunningStatus.Running,
+        workflow_tracing_count: 1,
+        workflow_data: workflowData,
+        workflow_data_json: JSON.stringify(workflowData),
+      },
     })
 
     mocks.mockTrackEvent.mockClear()
@@ -455,10 +457,12 @@ describe('useWorkflowRun', () => {
       workflow_id: 'flow-1',
       reason: 'Server Error',
       node_type: undefined,
-      workflow_status: WorkflowRunningStatus.Running,
-      workflow_tracing_count: 1,
-      workflow_data: workflowData,
-      workflow_data_json: JSON.stringify(workflowData),
+      data: {
+        workflow_status: WorkflowRunningStatus.Running,
+        workflow_tracing_count: 1,
+        workflow_data: workflowData,
+        workflow_data_json: JSON.stringify(workflowData),
+      },
     })
   })