diff --git a/src/frontend/src/modals/IOModal/components/IOFieldView/index.tsx b/src/frontend/src/modals/IOModal/components/IOFieldView/index.tsx
index 37c9461b5..ba8f83706 100644
--- a/src/frontend/src/modals/IOModal/components/IOFieldView/index.tsx
+++ b/src/frontend/src/modals/IOModal/components/IOFieldView/index.tsx
@@ -86,7 +86,7 @@ export default function IOFieldView({
                 value={
                   (flowPool[node.id] ?? [])[
                     (flowPool[node.id]?.length ?? 1) - 1
-                  ]?.params ?? ""
+                  ]?.data.results.result ?? ""
                 }
                 readOnly
               />
@@ -103,7 +103,7 @@ export default function IOFieldView({
                 value={
                   (flowPool[node.id] ?? [])[
                     (flowPool[node.id]?.length ?? 1) - 1
-                  ]?.params ?? ""
+                  ]?.data.results.result ?? ""
                 }
                 readOnly
               />
diff --git a/src/frontend/src/types/api/index.ts b/src/frontend/src/types/api/index.ts
index 4d73fe654..bcae42d51 100644
--- a/src/frontend/src/types/api/index.ts
+++ b/src/frontend/src/types/api/index.ts
@@ -159,7 +159,7 @@ export type VertexBuildTypeAPI = {
 // data is the object received by the API
 // it has results, artifacts, timedelta, duration
 export type VertexDataTypeAPI = {
-  results: { [key: string]: { [key: string]: string } };
+  results: { [key: string]:  string  };
   logs:{message:string}[];
   messages: ChatOutputType[] | chatInputType[];
   inactive?: boolean;
diff --git a/src/frontend/src/utils/reactflowUtils.ts b/src/frontend/src/utils/reactflowUtils.ts
index 4cd19bf7e..0eaca6961 100644
--- a/src/frontend/src/utils/reactflowUtils.ts
+++ b/src/frontend/src/utils/reactflowUtils.ts
@@ -1122,25 +1122,6 @@ export function expandGroupNode(
   setEdges(filteredEdges);
 }
 
-export function getGroupStatus(
-  flow: FlowType,
-  ssData: { [key: string]: { valid: boolean; params: string } }
-) {
-  let status = { valid: true, params: SUCCESS_BUILD };
-  const { nodes } = flow.data!;
-  const ids = nodes.map((n: NodeType) => n.data.id);
-  ids.forEach((id) => {
-    if (!ssData[id]) {
-      status = ssData[id];
-      return;
-    }
-    if (!ssData[id].valid) {
-      status = { valid: false, params: ssData[id].params };
-    }
-  });
-  return status;
-}
-
 export function createFlowComponent(
   nodeData: NodeDataType,
   version: string