diff --git a/wax-prosemirror-components/src/components/ImageUpload.js b/wax-prosemirror-components/src/components/ImageUpload.js
index b4a0ebfad89567024c4ec3a37914f2d3fb85d805..965f46be22f609676998bb98e9f8bf81a6acc9d1 100644
--- a/wax-prosemirror-components/src/components/ImageUpload.js
+++ b/wax-prosemirror-components/src/components/ImageUpload.js
@@ -1,5 +1,6 @@
 /* eslint react/prop-types: 0 */
-import React from 'react';
+import React, { useContext } from 'react';
+import { WaxContext } from 'wax-prosemirror-core';
 import styled from 'styled-components';
 
 const UploadImage = styled.div`