diff --git a/tools/rw-heatmaps/pkg/chart/heatmap_grid.go b/tools/rw-heatmaps/pkg/chart/heatmap_grid.go
index 00412e31ee47..5355264e9f1f 100644
--- a/tools/rw-heatmaps/pkg/chart/heatmap_grid.go
+++ b/tools/rw-heatmaps/pkg/chart/heatmap_grid.go
@@ -20,13 +20,13 @@ type HeatMapGrid struct {
 	z    [][]float64 // The Z values should be arranged in a 2D slice.
 }
 
-// Len returns the number of elements in the grid.
+// Dims returns the number of elements in the grid.
 // It implements the plotter.GridXYZ interface.
 func (h *HeatMapGrid) Dims() (int, int) {
 	return len(h.x), len(h.y)
 }
 
-// ValueAt returns the value of a grid cell at (c, r).
+// Z returns the value of a grid cell at (c, r).
 // It implements the plotter.GridXYZ interface.
 func (h *HeatMapGrid) Z(c, r int) float64 {
 	return h.z[r][c]