diff --git a/awx/ui_next/src/components/CodeEditor/CodeEditor.test.jsx b/awx/ui_next/src/components/CodeEditor/CodeEditor.test.jsx
index 0af1599feb..6f834d9108 100644
--- a/awx/ui_next/src/components/CodeEditor/CodeEditor.test.jsx
+++ b/awx/ui_next/src/components/CodeEditor/CodeEditor.test.jsx
@@ -1,5 +1,5 @@
 import React from 'react';
-import { mount } from 'enzyme';
+import { mountWithContexts } from '../../../testUtils/enzymeHelpers';
 import CodeEditor from './CodeEditor';
 
 describe('CodeEditor', () => {
@@ -9,7 +9,7 @@ describe('CodeEditor', () => {
 
   it('should pass value and mode through to ace editor', () => {
     const onChange = jest.fn();
-    const wrapper = mount(
+    const wrapper = mountWithContexts(
       <CodeEditor value={'---\nfoo: bar'} onChange={onChange} mode="yaml" />
     );
     const aceEditor = wrapper.find('AceEditor');
@@ -20,7 +20,7 @@ describe('CodeEditor', () => {
 
   it('should trigger onChange prop', () => {
     const onChange = jest.fn();
-    const wrapper = mount(
+    const wrapper = mountWithContexts(
       <CodeEditor value="---" onChange={onChange} mode="yaml" />
     );
     const aceEditor = wrapper.find('AceEditor');
@@ -30,7 +30,7 @@ describe('CodeEditor', () => {
 
   it('should render in read only mode', () => {
     const onChange = jest.fn();
-    const wrapper = mount(
+    const wrapper = mountWithContexts(
       <CodeEditor value="---" onChange={onChange} mode="yaml" readOnly />
     );
     const aceEditor = wrapper.find('AceEditor');