diff --git a/wax-prosemirror-services/src/CodeBlockService/CodeBlockTool.js b/wax-prosemirror-services/src/CodeBlockService/CodeBlockTool.js
index aeb128b36948db083d0f30d8df1f19aba248499c..6b2b9fca1fcd5bd68db6221ba8312a032084754d 100644
--- a/wax-prosemirror-services/src/CodeBlockService/CodeBlockTool.js
+++ b/wax-prosemirror-services/src/CodeBlockService/CodeBlockTool.js
@@ -1,6 +1,6 @@
 import { injectable } from 'inversify';
 import { icons } from 'wax-prosemirror-components';
-import { Commands } from 'wax-prosemirror-utilities';
+import { setBlockType } from 'prosemirror-commands';
 import Tools from '../lib/Tools';
 
 @injectable()
@@ -10,16 +10,13 @@ class CodeBlockTool extends Tools {
 
   get run() {
     return (state, dispatch) => {
-      Commands.setBlockType(state.config.schema.nodes.code_block)(
-        state,
-        dispatch,
-      );
+      setBlockType(state.config.schema.nodes.code_block)(state, dispatch);
     };
   }
 
   get enable() {
     return state => {
-      return Commands.setBlockType(state.config.schema.nodes.code_block)(state);
+      return setBlockType(state.config.schema.nodes.code_block)(state);
     };
   }
 }