Corrected nb_colors in heatmap