diff --git a/packages/fhir-flag/src/components/CloseFlagForm/tests/index.test.tsx b/packages/fhir-flag/src/components/CloseFlagForm/tests/index.test.tsx
index 80f30c2c2..b9dbc846a 100644
--- a/packages/fhir-flag/src/components/CloseFlagForm/tests/index.test.tsx
+++ b/packages/fhir-flag/src/components/CloseFlagForm/tests/index.test.tsx
@@ -5,6 +5,7 @@ import { flag } from '../../Utils/tests/fixtures';
import { QueryClient, QueryClientProvider } from 'react-query';
import userEvents from '@testing-library/user-event';
import { status } from '../../../constants';
+import { ContextProvider } from '@opensrp/react-utils';
jest.mock('@opensrp/notifications', () => ({
__esModule: true,
@@ -46,7 +47,9 @@ describe('CloseFlagForm component', () => {
render(
-
+
+
+
);
@@ -83,7 +86,9 @@ describe('CloseFlagForm component', () => {
render(
-
+
+
+
);
diff --git a/packages/fhir-flag/src/components/LocationFlag/tests/index.test.tsx b/packages/fhir-flag/src/components/LocationFlag/tests/index.test.tsx
index 2b94d2544..cfa3d938e 100644
--- a/packages/fhir-flag/src/components/LocationFlag/tests/index.test.tsx
+++ b/packages/fhir-flag/src/components/LocationFlag/tests/index.test.tsx
@@ -31,6 +31,8 @@ import {
import userEvents from '@testing-library/user-event';
import { status } from '../../../constants';
import * as notifications from '@opensrp/notifications';
+import { RoleContext } from '@opensrp/rbac';
+import { superUserRole } from '@opensrp/react-utils';
jest.mock('@opensrp/notifications', () => ({
__esModule: true,
@@ -122,9 +124,11 @@ test('renders correctly and fetches data', async () => {
render(
-
-
-
+
+
+
+
+
);