diff --git a/public/app/plugins/datasource/grafana-pyroscope-datasource/QueryEditor/QueryEditor.test.tsx b/public/app/plugins/datasource/grafana-pyroscope-datasource/QueryEditor/QueryEditor.test.tsx
index 75c0f98e564..643089f5f0e 100644
--- a/public/app/plugins/datasource/grafana-pyroscope-datasource/QueryEditor/QueryEditor.test.tsx
+++ b/public/app/plugins/datasource/grafana-pyroscope-datasource/QueryEditor/QueryEditor.test.tsx
@@ -111,6 +111,10 @@ function setupDs() {
     },
   ] as ProfileTypeMessage[]);
 
+  ds.getLabelNames = jest.fn().mockResolvedValue([
+      'label_one'
+  ]);
+
   return ds;
 }