Petit fix heatmap