diff --git a/ui/src/pages/Configuration/tests/ConfigContext.test.tsx b/ui/src/pages/Configuration/tests/ConfigContext.test.tsx
index b75c60a3b..9b8bff540 100644
--- a/ui/src/pages/Configuration/tests/ConfigContext.test.tsx
+++ b/ui/src/pages/Configuration/tests/ConfigContext.test.tsx
@@ -15,9 +15,9 @@ jest.mock('@splunk/search-job', () => ({
     create: () => ({
         getResults: () => ({
             subscribe: (
-                callbackFunction: (params: { results: { product_type: string }[] }) => void
+                callbackFunction: (params: { results: { instance_type: string }[] }) => void
             ) => {
-                callbackFunction({ results: [{ product_type: 'cloud' }] });
+                callbackFunction({ results: [{ instance_type: 'cloud' }] });
                 return { unsubscribe: () => {} };
             },
         }),
diff --git a/ui/src/pages/Input/tests/InputContext.test.tsx b/ui/src/pages/Input/tests/InputContext.test.tsx
index bfb753d81..e24c2c326 100644
--- a/ui/src/pages/Input/tests/InputContext.test.tsx
+++ b/ui/src/pages/Input/tests/InputContext.test.tsx
@@ -11,9 +11,9 @@ jest.mock('@splunk/search-job', () => ({
     create: () => ({
         getResults: () => ({
             subscribe: (
-                callbackFunction: (params: { results: { product_type: string }[] }) => void
+                callbackFunction: (params: { results: { instance_type: string }[] }) => void
             ) => {
-                callbackFunction({ results: [{ product_type: 'cloud' }] });
+                callbackFunction({ results: [{ instance_type: 'cloud' }] });
                 return { unsubscribe: () => {} };
             },
         }),