diff --git a/ui/src/components/SampleView/DrawGridPlugin.jsx b/ui/src/components/SampleView/DrawGridPlugin.jsx
index 334f5068b..f54bb1545 100644
--- a/ui/src/components/SampleView/DrawGridPlugin.jsx
+++ b/ui/src/components/SampleView/DrawGridPlugin.jsx
@@ -430,7 +430,7 @@ export default class DrawGridPlugin {
           if (result[index] !== undefined) {
             fillingMatrix[nw][nh] = this.heatMapColorForValue(
               gd,
-              result[index],
+              result[index][0],
             );
           }
         }