diff --git a/api/src/engine/models/result/heat-map-result.model.ts b/api/src/engine/models/result/heat-map-result.model.ts
index 267dc3c20201236e5d11ee1cbcbb6d6d89af15a7..8dd65ada27662536bffa7aeea6c8b78b0759c2fa 100644
--- a/api/src/engine/models/result/heat-map-result.model.ts
+++ b/api/src/engine/models/result/heat-map-result.model.ts
@@ -7,12 +7,12 @@ export class HeatMapResult extends Result {
   @Field()
   name: string;
 
-  @Field(() => [[Number]])
-  matrix: number[][];
+  @Field(() => ChartAxis, { nullable: true })
+  xAxis?: ChartAxis;
 
-  @Field(() => ChartAxis)
-  xAxis: ChartAxis;
+  @Field(() => ChartAxis, { nullable: true })
+  yAxis?: ChartAxis;
 
-  @Field(() => ChartAxis)
-  yAxis: ChartAxis;
+  @Field(() => [[Number]])
+  matrix: number[][];
 }