Correct heatmap scale