heatmap 2 fix