diff --git a/wax-prosemirror-services/src/AiService/components/ToggleAiComponent.js b/wax-prosemirror-services/src/AiService/components/ToggleAiComponent.js
index 9e6ca37edd5489b2477a0965003dc2aa1a57bd4a..6575a79f97262bbe6a17c86cd3b80a347266cdd2 100644
--- a/wax-prosemirror-services/src/AiService/components/ToggleAiComponent.js
+++ b/wax-prosemirror-services/src/AiService/components/ToggleAiComponent.js
@@ -1,4 +1,4 @@
-import React, { useContext, useMemo, useState } from 'react';
+import React, { useContext, useEffect, useMemo, useState } from 'react';
 import { WaxContext, MenuButton } from 'wax-prosemirror-core';
 import PropTypes from 'prop-types';
 
@@ -26,6 +26,12 @@ const ToggleAiComponent = ({ item }) => {
     main.focus();
   };
 
+  useEffect(() => {
+    setChecked(false);
+    context.setOption({ AiOn: false });
+    main.dispatch(main.state.tr.setMeta('addToHistory', false));
+  }, [checked && main.state.selection.from === main.state.selection.to]);
+
   return useMemo(
     () =>
       enableService.AiOn ? (