diff --git a/wax-prosemirror-core/src/ComponentPlugin.js b/wax-prosemirror-core/src/ComponentPlugin.js
index ca1f7c23a5062e630b766a884bb79d726b9285e7..805d667031930614e04ac937a7262bf0fb1746d5 100644
--- a/wax-prosemirror-core/src/ComponentPlugin.js
+++ b/wax-prosemirror-core/src/ComponentPlugin.js
@@ -1,5 +1,5 @@
 import React, { Component } from 'react';
-import { useInjection } from 'wax-prosemirror-core';
+import { useInjection } from './WaxContext';
 
 class UpdateView extends Component {
   constructor(props) {