diff --git a/wax-prosemirror-services/src/AiService/AskAiContentService.js b/wax-prosemirror-services/src/AiService/AskAiContentService.js
index 178253d34439368bbd2ec96d83d46b9e7d612b15..6f79d22a051222af24397aeb5bf24be297995f73 100644
--- a/wax-prosemirror-services/src/AiService/AskAiContentService.js
+++ b/wax-prosemirror-services/src/AiService/AskAiContentService.js
@@ -1,8 +1,8 @@
 import { Service } from 'wax-prosemirror-core';
 import AskAIOverlay from './components/AskAIOverlay';
 import AskAiSelectionPlugin from './plugins/AskAiSelectionPlugin';
-// import AiToolGroupService from './AiToolGroupService/AiToolGroupService';
-// import ToggleAiTool from './ToggleAiTool';
+import AiToolGroupService from './AiToolGroupService/AiToolGroupService';
+import ToggleAiTool from './ToggleAiTool';
 import './AskAiContent.css';
 
 class AskAiContentService extends Service {
@@ -31,10 +31,10 @@ class AskAiContentService extends Service {
   }
 
   register() {
-    // this.container.bind('ToggleAiTool').to(ToggleAiTool);
+    this.container.bind('ToggleAiTool').to(ToggleAiTool);
   }
 
-  // dependencies = [new AiToolGroupService()];
+  dependencies = [new AiToolGroupService()];
 }
 
 export default AskAiContentService;
diff --git a/wax-prosemirror-services/src/AiService/plugins/AskAiSelectionPlugin.js b/wax-prosemirror-services/src/AiService/plugins/AskAiSelectionPlugin.js
index 121981709ae853fdbba9bbb2cc1a7b1c088f3ae9..56ffbe9f184a1b7705df430cd9319e1d4355f0b5 100644
--- a/wax-prosemirror-services/src/AiService/plugins/AskAiSelectionPlugin.js
+++ b/wax-prosemirror-services/src/AiService/plugins/AskAiSelectionPlugin.js
@@ -14,14 +14,10 @@ export default () => {
         let createDecoration;
         const askAiInput = document.getElementById('askAiInput');
         if (askAiInput) {
-          const selectionWhenBlured = tr.getMeta(key);
+          // const selectionWhenBlured = tr.getMeta(key);
 
-          const from = selectionWhenBlured
-            ? selectionWhenBlured.from
-            : newState.selection.from;
-          const to = selectionWhenBlured
-            ? selectionWhenBlured.to
-            : newState.selection.to;
+          const from = newState.selection.from;
+          const to = newState.selection.to;
 
           createDecoration = DecorationSet.create(newState.doc, [
             Decoration.inline(from, to, {