fixes for heatmap