diff --git a/src/plugins/vis_default_editor/public/components/agg_params_helper.test.ts b/src/plugins/vis_default_editor/public/components/agg_params_helper.test.ts
index ecae5b511dff5..834ad8b70ad0d 100644
--- a/src/plugins/vis_default_editor/public/components/agg_params_helper.test.ts
+++ b/src/plugins/vis_default_editor/public/components/agg_params_helper.test.ts
@@ -133,6 +133,7 @@ describe('DefaultEditorAggParams helpers', () => {
             { name: '@timestamp', type: 'date' },
             { name: 'geo_desc', type: 'string' },
           ],
+          getAggregationRestrictions: jest.fn(),
         })),
         params: {
           orderBy: 'orderBy',