diff --git a/plugins/register-component/src/components/RegisterComponentPage/RegisterComponentPage.test.tsx b/plugins/register-component/src/components/RegisterComponentPage/RegisterComponentPage.test.tsx
index ad47c3a41b..ca463550d1 100644
--- a/plugins/register-component/src/components/RegisterComponentPage/RegisterComponentPage.test.tsx
+++ b/plugins/register-component/src/components/RegisterComponentPage/RegisterComponentPage.test.tsx
@@ -30,6 +30,8 @@ const catalogApi: jest.Mocked<typeof catalogApiRef.T> = {
   getEntities: jest.fn(),
   getEntityByName: jest.fn(),
   getLocationByEntity: jest.fn(),
+  getLocationById: jest.fn(),
+  getEntitiesByLocationId: jest.fn(),
 };
 
 const setup = () => ({