legend fix for treemap