diff --git a/public/app/plugins/datasource/testdata/datasource.ts b/public/app/plugins/datasource/testdata/datasource.ts
index 739e7bdade6..f364479d35b 100644
--- a/public/app/plugins/datasource/testdata/datasource.ts
+++ b/public/app/plugins/datasource/testdata/datasource.ts
@@ -95,7 +95,7 @@ export class TestDataDataSource extends DataSourceWithBackend<TestData> {
           streams.push(this.nodesQuery(target, options));
           break;
         case 'flame_graph':
-          streams.push(this.flameGraphQuery());
+          streams.push(this.flameGraphQuery(target));
           break;
         case 'trace':
           streams.push(this.trace(target, options));
@@ -242,8 +242,8 @@ export class TestDataDataSource extends DataSourceWithBackend<TestData> {
     return of({ data: frames }).pipe(delay(100));
   }
 
-  flameGraphQuery(): Observable<DataQueryResponse> {
-    return of({ data: [flameGraphData] }).pipe(delay(100));
+  flameGraphQuery(target: TestData): Observable<DataQueryResponse> {
+    return of({ data: [{ ...flameGraphData, refId: target.refId }] }).pipe(delay(100));
   }
 
   trace(target: TestData, options: DataQueryRequest<TestData>): Observable<DataQueryResponse> {
diff --git a/public/app/plugins/datasource/testdata/testData/flameGraphResponse.ts b/public/app/plugins/datasource/testdata/testData/flameGraphResponse.ts
index 1e4045dc636..370cf3ea5ff 100644
--- a/public/app/plugins/datasource/testdata/testData/flameGraphResponse.ts
+++ b/public/app/plugins/datasource/testdata/testData/flameGraphResponse.ts
@@ -2,7 +2,6 @@ import { DataFrameDTO } from '@grafana/data';
 
 export const flameGraphData: DataFrameDTO = {
   name: 'response',
-  refId: 'A',
   // @ts-ignore
   meta: { preferredVisualisationType: 'flamegraph' },
   fields: [