fixed color bug in heatmap