fixed heatmap's bug