Fixing the colors for the treemap #2