Select signals breaks legend naming

[mic3030]

Not sure how to fix the legend naming problem I am seeing..  The best way to see what is happening is to run the attached example. .  If you turn on all the signals and then turn off Triangle wave, the legend reads

Sine

Square

Sawtooth

DC

DC

There is only one DC signal but because the Triangle signal was turned off and the legend was expanded, it moves DC up in the list but also leaves it where it was, thus having two DC legends...   Any ideas how to fix this? 

Thanks

Mic

[AdnanZ]

You could use a property node. Attached is a modified version of your code. Remember, zero is not a vaild input to the property node. Let me know if you have any questions.