more fixing heatmap