diff --git a/src/frontend/src/stores/flowStore.ts b/src/frontend/src/stores/flowStore.ts
index 16f6b75f4..5ca1de1e0 100644
--- a/src/frontend/src/stores/flowStore.ts
+++ b/src/frontend/src/stores/flowStore.ts
@@ -473,7 +473,7 @@ const useFlowStore = create<FlowStoreType>((set, get) => ({
           runId: runId,
         });
         get().updateBuildStatus(
-          vertexBuildData.next_vertices_ids,
+          vertexBuildData.top_level_vertices,
           BuildStatus.TO_BUILD
         );
       }
diff --git a/src/frontend/src/types/api/index.ts b/src/frontend/src/types/api/index.ts
index 172f01dad..297984d28 100644
--- a/src/frontend/src/types/api/index.ts
+++ b/src/frontend/src/types/api/index.ts
@@ -146,6 +146,7 @@ export type VertexBuildTypeAPI = {
   id: string;
   inactivated_vertices: Array<string> | null;
   next_vertices_ids: Array<string>;
+  top_level_vertices: Array<string>;
   run_id: string;
   valid: boolean;
   params: string;