From 3f43958f089909beb8e9d38065139b5c58ad5f48 Mon Sep 17 00:00:00 2001
From: anovazzi1 <otavio2204@gmail.com>
Date: Tue, 28 May 2024 08:43:48 -0300
Subject: [PATCH] fix dropdown bug

---
 .../genericNode/components/OutputComponent/index.tsx           | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/frontend/src/customNodes/genericNode/components/OutputComponent/index.tsx b/src/frontend/src/customNodes/genericNode/components/OutputComponent/index.tsx
index d76f30f63..69ca3965c 100644
--- a/src/frontend/src/customNodes/genericNode/components/OutputComponent/index.tsx
+++ b/src/frontend/src/customNodes/genericNode/components/OutputComponent/index.tsx
@@ -10,6 +10,7 @@ import {
   DropdownMenuItem,
   DropdownMenuTrigger,
 } from "../../../../components/ui/dropdown-menu";
+import { useUpdateNodeInternals } from "reactflow";
 
 export default function OutputComponent({
   selected,
@@ -19,6 +20,7 @@ export default function OutputComponent({
   idx,
 }: outputComponentType) {
   const setNode = useFlowStore((state) => state.setNode);
+  const updateNodeInternals = useUpdateNodeInternals();
 
   if (types.length < 2) {
     return <span className={cn(frozen ? " text-ice" : "")}>{selected}</span>;
@@ -45,6 +47,7 @@ export default function OutputComponent({
                   type;
                 return newNode;
               });
+              updateNodeInternals(nodeId);
             }}
           >
             {type}