DrawViewportWires display update when Input parameter is removed

Hi,

In a custom component I override the DrawViewportWires method and everything works pretty good.

public override void DrawViewportWires(IGH_PreviewArgs args)
{
base.DrawViewportWires(args);
if (Hidden) { return; }
if (Locked) { return; }
....

}

...

If all my input parameter are set the code executes and the display modification are working well. If I hide or lock the component,  the display modification disappears - great...

But if I remove one of the input parameter, the display modification are still display. I am looking for a way to refresh the display of the component when input parameter is removed/changed.

thanks