diff --git a/src/components/nodes/LLMGenerateNode.tsx b/src/components/nodes/LLMGenerateNode.tsx
index c4e05614..523cb5fe 100644
--- a/src/components/nodes/LLMGenerateNode.tsx
+++ b/src/components/nodes/LLMGenerateNode.tsx
@@ -108,6 +108,8 @@ export function LLMGenerateNode({ id, data, selected }: NodeProps<LLMGenerateNod
       selected={selected}
       hasError={nodeData.status === "error"}
       commentNavigation={commentNavigation ?? undefined}
+      onRun={handleRegenerate}
+      isExecuting={isRunning}
     >
       {/* Image input - optional */}
       <Handle