heatmap classes fix