Fix heatmap