Wallacei - legend updating automatically problem

Hello people! Hope someone can clarify a little bit, what am I actually doing wrong here.

I am trying to shuffle through phenotypes with tree branch , but I cannot go through automatic updating of the legend with displacements from Karamba.
I guess I am not picking the tree items correctly to send them to the legend component input.

Hope someone can help me with this, there is a similar topic but not active anymore.

Thank you in advance!
Wallacei X_Karamba_ExampleFile_Distribute Legend to Grid (1).gh (58.6 KB)