diff --git a/d2/src/main/java/com/linkedin/d2/balancer/servers/ZooKeeperAnnouncer.java b/d2/src/main/java/com/linkedin/d2/balancer/servers/ZooKeeperAnnouncer.java index 4aa73a681..0a85e8547 100644 --- a/d2/src/main/java/com/linkedin/d2/balancer/servers/ZooKeeperAnnouncer.java +++ b/d2/src/main/java/com/linkedin/d2/balancer/servers/ZooKeeperAnnouncer.java @@ -176,7 +176,8 @@ public ZooKeeperAnnouncer(LoadBalancerServer server, boolean initialIsUp, _executorService = executorService; _eventEmitter = eventEmitter; - if(server instanceof ZooKeeperServer){ + if (server instanceof ZooKeeperServer) + { ((ZooKeeperServer) server).setServiceDiscoveryEventHelper(this); } }