BUGFIX: finxed heatmap