diff --git a/src/frontend/src/components/tableComponent/components/tableNodeCellRender/index.tsx b/src/frontend/src/components/tableComponent/components/tableNodeCellRender/index.tsx
index a4cf80ecc..fb24786f7 100644
--- a/src/frontend/src/components/tableComponent/components/tableNodeCellRender/index.tsx
+++ b/src/frontend/src/components/tableComponent/components/tableNodeCellRender/index.tsx
@@ -28,7 +28,7 @@ export default function TableNodeCellRender({
     value,
     nodeClass,
     handleOnNewValue: handleOnNewValueNode,
-    handleOnChangeDb,
+    handleOnChangeDb: handleOnChangeDbNode,
   },
 }: CustomCellRendererProps) {
   const handleOnNewValue = (newValue: any, name: string) => {
@@ -41,6 +41,15 @@ export default function TableNodeCellRender({
     setTemplateValue(newValue);
   };
 
+  const handleOnChangeDb = (newValue: boolean, name: string) => {
+    handleOnChangeDbNode(newValue, name);
+    setTemplateData((old) => {
+      let newData = cloneDeep(old);
+      newData.load_from_db = newValue;
+      return newData;
+    });
+  };
+
   const [templateValue, setTemplateValue] = useState(value);
   const [templateData, setTemplateData] = useState(data);
 
diff --git a/src/frontend/src/modals/editNodeModal/hooks/use-column-defs.tsx b/src/frontend/src/modals/editNodeModal/hooks/use-column-defs.tsx
index daddc5f1c..14263a9c6 100644
--- a/src/frontend/src/modals/editNodeModal/hooks/use-column-defs.tsx
+++ b/src/frontend/src/modals/editNodeModal/hooks/use-column-defs.tsx
@@ -6,6 +6,7 @@ import TableToggleCellRender from "../../../components/tableComponent/components
 const useColumnDefs = (
   myData: any,
   handleOnNewValue: (newValue: any, name: string) => void,
+  handleOnChangeDb: (value: boolean, key: string) => void,
   changeAdvanced: (n: string) => void,
   open: boolean,
 ) => {
@@ -47,9 +48,7 @@ const useColumnDefs = (
             value: params.data.value,
             nodeClass: myData.node,
             handleOnNewValue: handleOnNewValue,
-            handleOnChangeDb: (value, key) => {
-              myData.node!.template[key].load_from_db = value;
-            },
+            handleOnChangeDb: handleOnChangeDb,
           };
         },
         minWidth: 340,
diff --git a/src/frontend/src/modals/editNodeModal/index.tsx b/src/frontend/src/modals/editNodeModal/index.tsx
index 6394e85ce..b8e6a11a5 100644
--- a/src/frontend/src/modals/editNodeModal/index.tsx
+++ b/src/frontend/src/modals/editNodeModal/index.tsx
@@ -39,9 +39,12 @@ const EditNodeModal = forwardRef(
         !myData.current.node!.template[n]?.advanced;
     }
 
-    const handleOnNewValue = (newValue: any, name) => {
-      console.log(newValue);
-      myData.current.node!.template[name].value = newValue;
+    const handleOnNewValue = (newValue: any, key: string) => {
+      myData.current.node!.template[key].value = newValue;
+    };
+
+    const handleOnChangeDb = (newValue: boolean, key: string) => {
+      myData.current.node!.template[key].load_from_db = newValue;
     };
 
     const rowData = useRowData(data, open);
@@ -49,6 +52,7 @@ const EditNodeModal = forwardRef(
     const columnDefs: ColDef[] = useColumnDefs(
       data,
       handleOnNewValue,
+      handleOnChangeDb,
       changeAdvanced,
       open,
     );