Trigger saveWantedState when nodes are removed or orphaned wanted states are loaded

1 files changed, 2 insertions, 2 deletions

diff --git a/clustercontroller-core/src/test/java/com/yahoo/vespa/clustercontroller/core/StateChangeTest.java b/clustercontroller-core/src/test/java/com/yahoo/vespa/clustercontroller/core/StateChangeTest.java
index a5bb65e11d0..5a33414c955 100644
--- a/clustercontroller-core/src/test/java/com/yahoo/vespa/clustercontroller/core/StateChangeTest.java
+++ b/clustercontroller-core/src/test/java/com/yahoo/vespa/clustercontroller/core/StateChangeTest.java
@@ -1294,7 +1294,7 @@ public class StateChangeTest extends FleetControllerTest {
             NodeState newNodeState = new NodeState(NodeType.STORAGE, State.MAINTENANCE);
             NodeInfo nodeInfo = ctx.cluster.getNodeInfo(new Node(NodeType.STORAGE, 0));
             nodeInfo.setWantedState(newNodeState);
-            ctx.nodeStateOrHostInfoChangeHandler.handleNewWantedNodeState(nodeInfo, newNodeState);
+            ctx.nodeListener.handleNewWantedNodeState(nodeInfo, newNodeState);
             invoked = true;
         }
     }
@@ -1312,7 +1312,7 @@ public class StateChangeTest extends FleetControllerTest {
             NodeState newNodeState = new NodeState(NodeType.STORAGE, State.DOWN);
             NodeInfo nodeInfo = ctx.cluster.getNodeInfo(new Node(NodeType.STORAGE, 0));
             nodeInfo.setWantedState(newNodeState);
-            ctx.nodeStateOrHostInfoChangeHandler.handleNewWantedNodeState(nodeInfo, newNodeState);
+            ctx.nodeListener.handleNewWantedNodeState(nodeInfo, newNodeState);
             invoked = true;
         }
     }