diff --git a/plugins/register-component/src/components/RegisterComponentPage/RegisterComponentPage.test.tsx b/plugins/register-component/src/components/RegisterComponentPage/RegisterComponentPage.test.tsx
index d35691eadc..ad47c3a41b 100644
--- a/plugins/register-component/src/components/RegisterComponentPage/RegisterComponentPage.test.tsx
+++ b/plugins/register-component/src/components/RegisterComponentPage/RegisterComponentPage.test.tsx
@@ -29,6 +29,7 @@ const catalogApi: jest.Mocked<typeof catalogApiRef.T> = {
   addLocation: jest.fn((_a, _b) => new Promise(() => {})),
   getEntities: jest.fn(),
   getEntityByName: jest.fn(),
+  getLocationByEntity: jest.fn(),
 };
 
 const setup = () => ({