diff --git a/Scripts/Editor/NodeEditorWindow.cs b/Scripts/Editor/NodeEditorWindow.cs
index d626130..db16033 100644
--- a/Scripts/Editor/NodeEditorWindow.cs
+++ b/Scripts/Editor/NodeEditorWindow.cs
@@ -21,6 +21,7 @@ namespace XNodeEditor {
 
         void OnFocus() {
             current = this;
+            graphEditor = NodeGraphEditor.GetEditor(graph);
             if (NodeEditorPreferences.GetSettings().autoSave) AssetDatabase.SaveAssets();
         }