fix to heatmap if only one group