From 63e8e927e7505865e230ade427d226f44d9433bc Mon Sep 17 00:00:00 2001
From: anovazzi1 <otavio2204@gmail.com>
Date: Tue, 11 Jul 2023 15:29:48 -0300
Subject: [PATCH] fix(parameterComponent): add null check before calling
 cleanEdges to prevent error when reactFlowInstance is null

---
 .../components/parameterComponent/index.tsx      | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
index 2c94aa205..a3e3e6357 100644
--- a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
@@ -311,13 +311,15 @@ export default function ParameterComponent({
               field_name={name}
               setNodeClass={(nodeClass) => {
                 data.node = nodeClass;
-                cleanEdges({
-                  flow: {
-                    edges: reactFlowInstance.getEdges(),
-                    nodes: reactFlowInstance.getNodes(),
-                  },
-                  updateEdge: (edge) => reactFlowInstance.setEdges(edge),
-                });
+                if (reactFlowInstance) {
+                  cleanEdges({
+                    flow: {
+                      edges: reactFlowInstance.getEdges(),
+                      nodes: reactFlowInstance.getNodes(),
+                    },
+                    updateEdge: (edge) => reactFlowInstance.setEdges(edge),
+                  });
+                }
               }}
               nodeClass={data.node}
               disabled={disabled}