diff --git a/src/components/CodeEditor.tsx b/src/components/CodeEditor.tsx
index f709ce8..a4e3c96 100644
--- a/src/components/CodeEditor.tsx
+++ b/src/components/CodeEditor.tsx
@@ -1,7 +1,7 @@
 import { useState, useCallback, useEffect, useRef } from 'react';
-import { useStore } from 'zustand';
+
 import { useFlowStore } from '../store';
-import { useDiagramStore } from '../store/diagramStore';
+
 import { parseMermaid, detectInputType } from '../lib/mermaidParser';
 import { getLayoutedElements } from '../lib/layoutEngine';
 import { interpretText } from '../lib/aiService';
diff --git a/src/components/editor/EditorHeader.tsx b/src/components/editor/EditorHeader.tsx
index 1e9cfaa..ad06e88 100644
--- a/src/components/editor/EditorHeader.tsx
+++ b/src/components/editor/EditorHeader.tsx
@@ -20,7 +20,7 @@ export function EditorHeader() {
         theme, toggleTheme,
         focusMode, setFocusMode,
         saveDiagram,
-        aiMode, onlineProvider
+        aiMode
     } = useFlowStore();
     const { isMobile } = useMobileDetect();