From 12459dbdb9a0cefc8630fd4f57582e69d7ef703d Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Mon, 4 Mar 2024 10:55:32 -0300
Subject: [PATCH] Refactor validateNodes to use nodes from useFlowStore

---
 src/frontend/src/utils/buildUtils.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/frontend/src/utils/buildUtils.ts b/src/frontend/src/utils/buildUtils.ts
index e631c91de..ba3327959 100644
--- a/src/frontend/src/utils/buildUtils.ts
+++ b/src/frontend/src/utils/buildUtils.ts
@@ -125,7 +125,8 @@ export async function buildVertices({
 
   if (validateNodes) {
     try {
-      validateNodes(verticesIds);
+      const nodes = useFlowStore.getState().nodes;
+      validateNodes(nodes.map((node) => node.id));
     } catch (e) {
       return;
     }