diff --git a/src/containers/Header/test/Search.test.js b/src/containers/Header/test/Search.test.js
index 9d9904435..f383a3f8f 100644
--- a/src/containers/Header/test/Search.test.js
+++ b/src/containers/Header/test/Search.test.js
@@ -1,6 +1,7 @@
 import { render, screen, cleanup, fireEvent } from '@testing-library/react'
 import { I18nextProvider } from 'react-i18next'
 import { BrowserRouter as Router } from 'react-router-dom'
+import { QueryClientProvider } from 'react-query'
 import i18n from '../../../i18n/testConfig'
 import { Search } from '../Search'
 import * as rippled from '../../../rippled/lib/rippled'
diff --git a/src/containers/shared/components/test/DomainLink.test.tsx b/src/containers/shared/components/test/DomainLink.test.tsx
index 020323836..cb42c506e 100644
--- a/src/containers/shared/components/test/DomainLink.test.tsx
+++ b/src/containers/shared/components/test/DomainLink.test.tsx
@@ -67,9 +67,11 @@ describe('DomainLink', () => {
 
   it('handles domain link with protocol removal', () => {
     const url = 'https://example.com/'
-    const wrapper = mount(<DomainLink domain={url} keepProtocol={false} />)
-    expect(wrapper.find('a').props().className).toEqual('domain')
-    expect(wrapper.find('a').text()).toEqual('example.com')
-    expect(wrapper.find('a').props().href).toEqual('https://example.com/')
+    const { container } = render(
+      <DomainLink domain={url} keepProtocol={false} />,
+    )
+    expect(container).toHaveTextContent('example.com')
+    expect(screen.getByText('example.com')).toHaveAttribute('href', url)
+    expect(screen.getByText('example.com')).toHaveClass('domain')
   })
 })