Fix TreeHighlighter for restored connections
Fix LagPlugin for restored connections