diff --git a/src/services/llm/chat_service.ts b/src/services/llm/chat_service.ts
index f928847ac..ff6afc021 100644
--- a/src/services/llm/chat_service.ts
+++ b/src/services/llm/chat_service.ts
@@ -168,10 +168,10 @@ export class ChatService {
 
         if (useSmartContext && lastUserMessage) {
             // Use smart context that considers the query for better relevance
-            context = await contextExtractor.getSmartContext(noteId, lastUserMessage);
+            context = await aiServiceManager.getContextExtractor().getSmartContext(noteId, lastUserMessage);
         } else {
             // Fall back to full context if smart context is disabled or no query available
-            context = await contextExtractor.getFullContext(noteId);
+            context = await aiServiceManager.getContextExtractor().getFullContext(noteId);
         }
 
         const contextMessage: Message = {