Fix heatmap code granularity