diff --git a/plugins/allure/src/components/AllureReportComponent/AllureReportComponent.test.tsx b/plugins/allure/src/components/AllureReportComponent/AllureReportComponent.test.tsx
index 34d12aa032..e6355d861f 100644
--- a/plugins/allure/src/components/AllureReportComponent/AllureReportComponent.test.tsx
+++ b/plugins/allure/src/components/AllureReportComponent/AllureReportComponent.test.tsx
@@ -23,6 +23,7 @@ import {
setupRequestMockHandlers,
renderInTestApp,
} from '@backstage/test-utils';
+import { EntityProvider } from '@backstage/plugin-catalog-react';
describe('ExampleComponent', () => {
const server = setupServer();
@@ -39,7 +40,15 @@ describe('ExampleComponent', () => {
it('should render', async () => {
const rendered = await renderInTestApp(
-
+
+
+
,
);
expect(rendered.getByText('Missing Annotation')).toBeInTheDocument();
diff --git a/plugins/todo/src/plugin.test.tsx b/plugins/todo/src/plugin.test.tsx
index 57b9be291d..847ab7f65c 100644
--- a/plugins/todo/src/plugin.test.tsx
+++ b/plugins/todo/src/plugin.test.tsx
@@ -19,6 +19,7 @@ import { Route } from 'react-router';
import { renderInTestApp, TestApiProvider } from '@backstage/test-utils';
import { todoPlugin, EntityTodoContent } from './plugin';
import { todoApiRef } from './api';
+import { EntityProvider } from '@backstage/plugin-catalog-react';
describe('todo', () => {
it('should export plugin', () => {
@@ -47,7 +48,15 @@ describe('todo', () => {
],
]}
>
- } />
+
+ } />
+
,
);