heatmap small fix
heatmap small fix