diff --git a/src/frontend/src/CustomNodes/helpers/mutate-template.ts b/src/frontend/src/CustomNodes/helpers/mutate-template.ts
index 7aeab13f8..5fdfd3caa 100644
--- a/src/frontend/src/CustomNodes/helpers/mutate-template.ts
+++ b/src/frontend/src/CustomNodes/helpers/mutate-template.ts
@@ -6,6 +6,7 @@ import {
 import { APIClassType, ResponseErrorDetailAPI } from "@/types/api";
 import { UseMutationResult } from "@tanstack/react-query";
 import { cloneDeep, debounce } from "lodash";
+import { updateHiddenOutputs } from "./update-hidden-outputs";
 
 // Map to store debounced functions for each node ID
 const debouncedFunctions = new Map<string, ReturnType<typeof debounce>>();
@@ -53,7 +54,10 @@ export const mutateTemplate = async (
             });
             if (newTemplate) {
               newNode.template = newTemplate.template;
-              newNode.outputs = newTemplate.outputs;
+              newNode.outputs = updateHiddenOutputs(
+                newNode.outputs ?? [],
+                newTemplate.outputs ?? [],
+              );
               newNode.tool_mode = toolMode ?? node.tool_mode;
             }
             setNodeClass(newNode);