diff --git a/wax-prosemirror-components/src/components/images/ImageUpload.js b/wax-prosemirror-components/src/components/images/ImageUpload.js
index 300b60b924c33a5c0814aa281d09041d90d2773d..8d66bdda027f9671e1fee76e590009296b74d047 100644
--- a/wax-prosemirror-components/src/components/images/ImageUpload.js
+++ b/wax-prosemirror-components/src/components/images/ImageUpload.js
@@ -15,7 +15,7 @@ const Wrapper = styled.div`
 const ImageUpload = ({ item, fileUpload, view }) => {
   const {
     app,
-    activeViewId,
+    activeView,
     view: { main },
   } = useContext(WaxContext);
 
@@ -37,7 +37,7 @@ const ImageUpload = ({ item, fileUpload, view }) => {
     insertImage(urls, view, placeholderPlugin);
   }
 
-  let isDisabled = !item.select(view.state, activeViewId);
+  let isDisabled = !item.select(activeView);
 
   const isEditable = main.props.editable(editable => {
     return editable;
diff --git a/wax-prosemirror-components/src/components/tables/CreateTable.js b/wax-prosemirror-components/src/components/tables/CreateTable.js
index 78b1453da14c55c6c5abce20c0794a3e0200ab27..2ddcc4ecc1744a6c2b7a0e426f52e843fdafb5d3 100644
--- a/wax-prosemirror-components/src/components/tables/CreateTable.js
+++ b/wax-prosemirror-components/src/components/tables/CreateTable.js
@@ -27,9 +27,6 @@ const CreateTable = ({ view = {}, item }) => {
     activeView,
     activeViewId,
   } = useContext(WaxContext);
-  if (item.onlyOnMain) {
-    view = main;
-  }
 
   const { icon, run, select, title } = item;
   const ref = useRef();
@@ -48,7 +45,7 @@ const CreateTable = ({ view = {}, item }) => {
     setIsOpen(!isOpen);
   };
 
-  let isDisabled = !select(activeView, activeViewId);
+  let isDisabled = !select(activeView);
 
   const isEditable = main.props.editable(editable => {
     return editable;
diff --git a/wax-prosemirror-services/src/ImageService/Image.js b/wax-prosemirror-services/src/ImageService/Image.js
index 382f08e69bd8aebf8b5fca1d8f80fcc834d55ab1..a5e792d5d3007a1e2caf06a9c0281753a17d6fa5 100644
--- a/wax-prosemirror-services/src/ImageService/Image.js
+++ b/wax-prosemirror-services/src/ImageService/Image.js
@@ -18,8 +18,9 @@ class Image extends Tools {
     return true;
   }
 
-  select = (state, activeViewId) => {
-    if (activeViewId !== 'main') return false;
+  select = activeView => {
+    const { disallowedTools } = activeView.props;
+    if (disallowedTools.includes('images')) return false;
     return true;
   };
 
diff --git a/wax-prosemirror-services/src/NoteService/Editor.js b/wax-prosemirror-services/src/NoteService/Editor.js
index 21b868fcf6d3888c221f009892a031b0edabc433..42a65453dddd58ceedbea3edadefb4a729568ddd 100644
--- a/wax-prosemirror-services/src/NoteService/Editor.js
+++ b/wax-prosemirror-services/src/NoteService/Editor.js
@@ -37,7 +37,7 @@ export default ({ node, view }) => {
         }),
         // This is the magic part
         dispatchTransaction,
-        disallowedTools: ['tables'],
+        disallowedTools: ['tables', 'images'],
         handleDOMEvents: {
           blur: () => {
             if (context.view[noteId]) {
diff --git a/wax-prosemirror-services/src/TablesService/InsertTableService/Table.js b/wax-prosemirror-services/src/TablesService/InsertTableService/Table.js
index b3e6bf75b320e8d184e5f32a534007c961c9a7ab..2ac2d69abebdd94813fab04d316de7a511eda7fe 100644
--- a/wax-prosemirror-services/src/TablesService/InsertTableService/Table.js
+++ b/wax-prosemirror-services/src/TablesService/InsertTableService/Table.js
@@ -19,8 +19,8 @@ class Table extends Tools {
     };
   }
 
-  select = (view, activeViewId) => {
-    const { disallowedTools } = view.props;
+  select = activeView => {
+    const { disallowedTools } = activeView.props;
     if (disallowedTools.includes('tables')) return false;
     return true;
   };