heatmap classes fix 2