diff --git a/src/frontend/src/CustomNodes/hooks/use-check-code-validity.tsx b/src/frontend/src/CustomNodes/hooks/use-check-code-validity.tsx
index 0b83429f8..4cbfbd8df 100644
--- a/src/frontend/src/CustomNodes/hooks/use-check-code-validity.tsx
+++ b/src/frontend/src/CustomNodes/hooks/use-check-code-validity.tsx
@@ -26,7 +26,8 @@ const useCheckCodeValidity = (
     const componentsToIgnore = ["Custom Component", "Prompt"];
     if (
       currentCode !== thisNodesCode &&
-      !componentsToIgnore.includes(data.node!.display_name)
+      !componentsToIgnore.includes(data.node!.display_name) &&
+      !(data.node?.edited ?? false)
     ) {
       setIsOutdated(true);
     } else {
diff --git a/src/frontend/src/CustomNodes/hooks/use-update-node-code.tsx b/src/frontend/src/CustomNodes/hooks/use-update-node-code.tsx
index 28220d141..f1593597f 100644
--- a/src/frontend/src/CustomNodes/hooks/use-update-node-code.tsx
+++ b/src/frontend/src/CustomNodes/hooks/use-update-node-code.tsx
@@ -19,6 +19,7 @@ const useUpdateNodeCode = (
           node: newNodeClass,
           description: newNodeClass.description ?? dataNode.description,
           display_name: newNodeClass.display_name ?? dataNode.display_name,
+          edited: false,
         };
 
         newNode.data.node.template[name].value = code;