Adapt to changes in GraphVisualizer