diff --git a/cluster4npu_ui/core/functions/InferencePipeline.py b/cluster4npu_ui/core/functions/InferencePipeline.py
index f6bbd4d..63fec81 100644
--- a/cluster4npu_ui/core/functions/InferencePipeline.py
+++ b/cluster4npu_ui/core/functions/InferencePipeline.py
@@ -257,8 +257,8 @@ class PipelineStage:
                 inference_result = {'probability': 0.0, 'result': 'Processing', 'status': 'async'}
             
             # Step 4: Update pipeline data
-            pipeline_data.stage_results[self.stage_id] = processed_result
-            pipeline_data.data = processed_result  # Pass result as data to next stage
+            pipeline_data.stage_results[self.stage_id] = inference_result
+            pipeline_data.data = inference_result  # Pass result as data to next stage
             pipeline_data.metadata[f'{self.stage_id}_timestamp'] = time.time()
             
             return pipeline_data