diff --git a/src/backend/langflow/api/v1/chat.py b/src/backend/langflow/api/v1/chat.py
index fd0232189..25be65dbb 100644
--- a/src/backend/langflow/api/v1/chat.py
+++ b/src/backend/langflow/api/v1/chat.py
@@ -47,7 +47,7 @@ async def init_build(graph_data: dict, flow_id: str):
                 logger.debug(f"Deleted flow {flow_id} from cache")
         flow_data_store[flow_id] = {
             "graph_data": graph_data,
-            "status": BuildStatus.IN_PROGRESS,
+            "status": BuildStatus.STARTED,
         }
 
         return InitResponse(flowId=flow_id)
@@ -91,7 +91,7 @@ async def stream_build(flow_id: str):
                 yield str(StreamData(event="error", data={"error": error_message}))
                 return
 
-            graph_data = flow_data_store[flow_id].get("data")
+            graph_data = flow_data_store[flow_id].get("graph_data")
 
             if not graph_data:
                 error_message = "No data provided"
@@ -109,6 +109,7 @@ async def stream_build(flow_id: str):
                 return
 
             number_of_nodes = len(graph.nodes)
+            flow_data_store[flow_id]["status"] = BuildStatus.IN_PROGRESS
             for i, vertex in enumerate(graph.generator_build(), 1):
                 try:
                     log_dict = {
diff --git a/src/backend/langflow/api/v1/schemas.py b/src/backend/langflow/api/v1/schemas.py
index 2cf62a504..f5f1f9ccf 100644
--- a/src/backend/langflow/api/v1/schemas.py
+++ b/src/backend/langflow/api/v1/schemas.py
@@ -11,6 +11,7 @@ class BuildStatus(Enum):
 
     SUCCESS = "success"
     FAILURE = "failure"
+    STARTED = "started"
     IN_PROGRESS = "in_progress"