Hi Clark, it's an anoying thing for sure. What I've seen in this cases vary on how 'open mind' your customer is. For example you can try to use the tree map object without dimension, 1 as metric and use the same custom color expression. That way you will keep the assosiation and you can simulate a legend/filter.
Hope it helps,