diff --git a/editors/demo/src/Editoria/layout/EditoriaMobileLayout.js b/editors/demo/src/Editoria/layout/EditoriaMobileLayout.js
index c2df40828054b683f835d00124dbf994cc1872b6..b0fd63c353d526a34bde8d38c03c75afddaa7a69 100644
--- a/editors/demo/src/Editoria/layout/EditoriaMobileLayout.js
+++ b/editors/demo/src/Editoria/layout/EditoriaMobileLayout.js
@@ -202,7 +202,7 @@ const EditoriaLayout = ({ editor }) => {
 
   return (
     <ThemeProvider theme={cokoTheme}>
-      <Wrapper>
+      <Wrapper id="wax-container">
         <TopMenu>
           <TopBar />
         </TopMenu>
diff --git a/wax-prosemirror-services/src/NoteService/Editor.js b/wax-prosemirror-services/src/NoteService/Editor.js
index e0f812867b9c535ddf25d849fc7fc8efb0da91ee..a823668eb4fb122634821d76b30858f1735bc4e4 100644
--- a/wax-prosemirror-services/src/NoteService/Editor.js
+++ b/wax-prosemirror-services/src/NoteService/Editor.js
@@ -116,10 +116,13 @@ export default ({ node, view }) => {
     }, 20);
 
     const findReplace = context.app.PmPlugins.get('findAndReplacePlugin');
-    const matches = findReplace.getState(noteView.state).allMatches;
-    if (matches.length > 0 && !typing && context.activeViewId === noteId)
-      context.updateView({}, noteId);
-    // UNTIL HERE
+
+    if (findReplace) {
+      const matches = findReplace.getState(noteView.state).allMatches;
+      if (matches.length > 0 && !typing && context.activeViewId === noteId)
+        context.updateView({}, noteId);
+      // UNTIL HERE
+    }
 
     if (!tr.getMeta('fromOutside')) {
       const outerTr = view.state.tr;