diff --git a/wax-prosemirror-plugins/src/WaxSelectionPlugin.js b/wax-prosemirror-plugins/src/WaxSelectionPlugin.js
index 100f9c20ecbb20e05642eb74f57cbeca7e8b7e50..6be0677f0c82f7477a99d511efa9ea62961db116 100644
--- a/wax-prosemirror-plugins/src/WaxSelectionPlugin.js
+++ b/wax-prosemirror-plugins/src/WaxSelectionPlugin.js
@@ -9,13 +9,6 @@ const WaxSelectionPlugin = new Plugin({
     apply(transaction, state, prevEditorState, editorState) {
       const sel = transaction.curSelection;
 
-      // TODO fix the selection when a note is present.
-      let flag = false;
-      const difference = sel.$to.pos - sel.$from.pos;
-      editorState.doc.nodesBetween(sel.$from.pos, sel.$to.pos, (node, from) => {
-        if (node.type.name === 'footnote') flag = true;
-      });
-
       const decos = [
         Decoration.inline(sel.$from.pos, sel.$to.pos, {
           class: 'wax-selection-marker',
@@ -23,8 +16,6 @@ const WaxSelectionPlugin = new Plugin({
       ];
       const deco = DecorationSet.create(editorState.doc, decos);
       return { deco };
-
-      return state;
     },
   },
   props: {