diff --git a/webapp/channels/src/components/toggle_modal_button.test.tsx b/webapp/channels/src/components/toggle_modal_button.test.tsx
index cf487f6367..8b7a11247a 100644
--- a/webapp/channels/src/components/toggle_modal_button.test.tsx
+++ b/webapp/channels/src/components/toggle_modal_button.test.tsx
@@ -15,19 +15,15 @@ jest.mock('react-redux', () => ({
useDispatch: () => jest.fn(),
}));
-class TestModal extends React.PureComponent {
- render() {
- return (
-
-
-
-
- );
- }
-}
+const TestModal = () => (
+
+
+
+
+);
describe('components/ToggleModalButton', () => {
test('component should match snapshot', () => {