diff --git a/Scripts/Editor/NodeEditorGUI.cs b/Scripts/Editor/NodeEditorGUI.cs
index 7230121..cda5b98 100644
--- a/Scripts/Editor/NodeEditorGUI.cs
+++ b/Scripts/Editor/NodeEditorGUI.cs
@@ -364,12 +364,11 @@ namespace XNodeEditor {
                 //Draw node contents
                 nodeEditor.OnNodeGUI();
 
-                //Apply
-                nodeEditor.serializedObject.ApplyModifiedProperties();
-
                 //If user changed a value, notify other scripts through onUpdateNode
                 if (EditorGUI.EndChangeCheck()) {
                     if (NodeEditor.onUpdateNode != null) NodeEditor.onUpdateNode(node);
+                    EditorUtility.SetDirty(node);
+                    nodeEditor.serializedObject.ApplyModifiedProperties();
                 }
 
                 GUILayout.EndVertical();