diff --git a/wax-prosemirror-components/index.js b/wax-prosemirror-components/index.js
index bdcac1b0f4b894be109977070f1986461250fa9b..a34e71fe9f0d4d426dc9f12ca985459b84629a31 100644
--- a/wax-prosemirror-components/index.js
+++ b/wax-prosemirror-components/index.js
@@ -11,7 +11,3 @@ export { default as Tabs } from './src/ui/tabs/Tabs';
 export { default as BlockLevelTools } from './src/ui/tabs/BlockLevelTools';
 export { default as EditorInfoTool } from './src/components/EditorInfo/CounterInfo/EditorInfoTool';
 export { default as EditorShortCutsTool } from './src/components/EditorInfo/EditorShortCutsTool';
-
-export { default as CustomTagInlineComponent } from './src/components/customtag/CustomTagInlineComponent';
-export { default as CustomTagInlineOverlayComponent } from './src/components/customtag/CustomTagInlineOverlayCompoment';
-export { default as CustomTagBlockComponent } from './src/components/customtag/CustomTagBlockComponent';
diff --git a/wax-prosemirror-components/src/ui/tabs/BlockElementGroup.js b/wax-prosemirror-components/src/ui/tabs/BlockElementGroup.js
index f5a4ca81c8b80ada1321cfbaf84814ebf048497a..732742cf490d516b46619fde3178412c57766ec0 100644
--- a/wax-prosemirror-components/src/ui/tabs/BlockElementGroup.js
+++ b/wax-prosemirror-components/src/ui/tabs/BlockElementGroup.js
@@ -2,7 +2,7 @@
 import React, { useState } from 'react';
 import { th } from '@pubsweet/ui-toolkit';
 import styled from 'styled-components';
-import CustomTagBlockComponent from '../../components/customtag/CustomTagBlockComponent';
+import CustomTagBlockComponent from './CustomTagBlockComponent';
 import OENToolGroup from './OENToolGroup';
 import BlockElement from './BlockElement';
 import Icon from '../../helpers/Icon';
diff --git a/wax-prosemirror-components/src/components/customtag/CustomTagBlockComponent.js b/wax-prosemirror-components/src/ui/tabs/CustomTagBlockComponent.js
similarity index 98%
rename from wax-prosemirror-components/src/components/customtag/CustomTagBlockComponent.js
rename to wax-prosemirror-components/src/ui/tabs/CustomTagBlockComponent.js
index 2e1ddd8e915874880e63976d29728de4f4c39c11..e83e150b7efca084626ab32cd965b0ae361282f2 100644
--- a/wax-prosemirror-components/src/components/customtag/CustomTagBlockComponent.js
+++ b/wax-prosemirror-components/src/ui/tabs/CustomTagBlockComponent.js
@@ -2,7 +2,7 @@ import React, { useContext, useMemo, useRef, useState, useEffect } from 'react';
 import styled, { css } from 'styled-components';
 import { WaxContext } from 'wax-prosemirror-core';
 import { v4 as uuidv4 } from 'uuid';
-import MenuButton from '../../ui/buttons/MenuButton';
+import MenuButton from '../buttons/MenuButton';
 
 const activeStyles = css`
   pointer-events: none;
diff --git a/wax-prosemirror-services/src/CustomTagService/CustomTagInlineService/CustomTagInlineService.js b/wax-prosemirror-services/src/CustomTagService/CustomTagInlineService/CustomTagInlineService.js
index a7da291b3d62976df2a9a865118380a7d58fbb28..69c10228207ca01f1ccf7c0198558c1c155be88c 100644
--- a/wax-prosemirror-services/src/CustomTagService/CustomTagInlineService/CustomTagInlineService.js
+++ b/wax-prosemirror-services/src/CustomTagService/CustomTagInlineService/CustomTagInlineService.js
@@ -1,5 +1,5 @@
 import { Service } from 'wax-prosemirror-core';
-import { CustomTagInlineOverlayComponent } from 'wax-prosemirror-components';
+import CustomTagInlineOverlayComponent from '../components/CustomTagInlineOverlayComponent';
 import customtagInlineMark from './schema/customtagInlineMark';
 import CustomTagInlineTool from './CustomTagInlineTool';
 
diff --git a/wax-prosemirror-services/src/CustomTagService/CustomTagInlineService/CustomTagInlineTool.js b/wax-prosemirror-services/src/CustomTagService/CustomTagInlineService/CustomTagInlineTool.js
index d2fa2eff5e7b9079cbe1fca3392ce6eff56150ac..521a91c5e05fb928197d58127df9b275fa5eb068 100644
--- a/wax-prosemirror-services/src/CustomTagService/CustomTagInlineService/CustomTagInlineTool.js
+++ b/wax-prosemirror-services/src/CustomTagService/CustomTagInlineService/CustomTagInlineTool.js
@@ -1,7 +1,7 @@
 import React from 'react';
 import { v4 as uuidv4 } from 'uuid';
-import { CustomTagInlineComponent } from 'wax-prosemirror-components';
 import { Tools } from 'wax-prosemirror-core';
+import CustomTagInlineComponent from '../components/CustomTagInlineComponent';
 
 class CustomTagInLineTool extends Tools {
   title = 'Custom Tag Inline';
diff --git a/wax-prosemirror-components/src/components/customtag/CustomTagInlineComponent.js b/wax-prosemirror-services/src/CustomTagService/components/CustomTagInlineComponent.js
similarity index 95%
rename from wax-prosemirror-components/src/components/customtag/CustomTagInlineComponent.js
rename to wax-prosemirror-services/src/CustomTagService/components/CustomTagInlineComponent.js
index e85a8ee4bfe1f7d7597a42ef3dfe24880e14b5ea..ddbfb3ff7b7101c53cab83194ebd2edbf2d7236f 100644
--- a/wax-prosemirror-components/src/components/customtag/CustomTagInlineComponent.js
+++ b/wax-prosemirror-services/src/CustomTagService/components/CustomTagInlineComponent.js
@@ -1,8 +1,7 @@
 import React, { useMemo, useState, useContext } from 'react';
-import { WaxContext } from 'wax-prosemirror-core';
 import styled from 'styled-components';
-
-import MenuButton from '../../ui/buttons/MenuButton';
+import { WaxContext } from 'wax-prosemirror-core';
+import { MenuButton } from 'wax-prosemirror-components';
 
 const StyledButton = styled(MenuButton)`
   svg {
diff --git a/wax-prosemirror-components/src/components/customtag/CustomTagInlineOverlayCompoment.js b/wax-prosemirror-services/src/CustomTagService/components/CustomTagInlineOverlayComponent.js
similarity index 97%
rename from wax-prosemirror-components/src/components/customtag/CustomTagInlineOverlayCompoment.js
rename to wax-prosemirror-services/src/CustomTagService/components/CustomTagInlineOverlayComponent.js
index 61ff08bc318d4947018c03511532c79cdb0d2698..da3549006125e584017d58688471000984e5f622 100644
--- a/wax-prosemirror-components/src/components/customtag/CustomTagInlineOverlayCompoment.js
+++ b/wax-prosemirror-services/src/CustomTagService/components/CustomTagInlineOverlayComponent.js
@@ -10,7 +10,7 @@ import styled from 'styled-components';
 import { grid, th } from '@pubsweet/ui-toolkit';
 import { WaxContext } from 'wax-prosemirror-core';
 import { v4 as uuidv4 } from 'uuid';
-import Icon from '../../helpers/Icon';
+import { Icon } from 'wax-prosemirror-components';
 
 const IconRemove = styled(Icon)`
   cursor: pointer;
@@ -80,7 +80,7 @@ const Item = styled.div`
 
 const initialArr = [];
 
-const CustomTagInlineOverlayComponent = ({ mark, setPosition, position }) => {
+const CustomTagInlineOverlayComponent = ({ mark }) => {
   const ref = useRef(null);
 
   const [inputValue, setInputValue] = useState('');
@@ -111,7 +111,7 @@ const CustomTagInlineOverlayComponent = ({ mark, setPosition, position }) => {
   const [allTags, setAllTags] = useState(configTags);
 
   let image = false;
-  state.doc.nodesBetween($from.pos, $to.pos, (node, pos) => {
+  state.doc.nodesBetween($from.pos, $to.pos, node => {
     if (node.type.name === 'image') {
       image = true;
     }