diff --git a/src/courseware/course/sequence/sequence-navigation/UnitIcon.test.jsx b/src/courseware/course/sequence/sequence-navigation/UnitIcon.test.jsx
index 509474ef..8bfae84c 100644
--- a/src/courseware/course/sequence/sequence-navigation/UnitIcon.test.jsx
+++ b/src/courseware/course/sequence/sequence-navigation/UnitIcon.test.jsx
@@ -1,5 +1,5 @@
import React from 'react';
-import { render, screen } from '../../../../setupTest';
+import { render } from '../../../../setupTest';
import UnitIcon from './UnitIcon';
describe('Unit Icon', () => {
@@ -19,8 +19,8 @@ describe('Unit Icon', () => {
jest.spyOn(console, 'error').mockImplementation(() => {});
}
- render();
- expect(screen.getByTestId('icon')).toHaveClass(value);
+ const { container } = render();
+ expect(container.querySelector('svg')).toHaveClass(value);
});
});
});
diff --git a/src/test/__mocks__/@fortawesome/react-fontawesome.js b/src/test/__mocks__/@fortawesome/react-fontawesome.js
deleted file mode 100644
index 2d1ef1cc..00000000
--- a/src/test/__mocks__/@fortawesome/react-fontawesome.js
+++ /dev/null
@@ -1,32 +0,0 @@
-/**
- * Mocks `@fortawesome/react-fontawesome.js` to return a simple
element containing `data-testid` attribute.
- * This way we can check whether the icon matches without relying on its internal implementation
- * and avoid storing its