Sync with messagebus thread when disconnecting

2 files changed, 2 insertions, 2 deletions

diff --git a/messagebus/src/main/java/com/yahoo/messagebus/DestinationSession.java b/messagebus/src/main/java/com/yahoo/messagebus/DestinationSession.java
index 10d1a93031d..32f7ccd2e18 100644
--- a/messagebus/src/main/java/com/yahoo/messagebus/DestinationSession.java
+++ b/messagebus/src/main/java/com/yahoo/messagebus/DestinationSession.java
@@ -130,6 +130,6 @@ public final class DestinationSession implements Connectable, MessageHandler {
         mbus.connect(name, broadcastName);
     }
 
-    @Override public void disconnect() { mbus.unregisterSession(name, broadcastName);; }
+    @Override public void disconnect() { close(); }
 
 }
diff --git a/messagebus/src/main/java/com/yahoo/messagebus/IntermediateSession.java b/messagebus/src/main/java/com/yahoo/messagebus/IntermediateSession.java
index 1473f40ef0a..cbc5d89be79 100644
--- a/messagebus/src/main/java/com/yahoo/messagebus/IntermediateSession.java
+++ b/messagebus/src/main/java/com/yahoo/messagebus/IntermediateSession.java
@@ -119,6 +119,6 @@ public final class IntermediateSession implements MessageHandler, ReplyHandler,
         mbus.connect(name, broadcastName);
     }
 
-    @Override public void disconnect() { mbus.unregisterSession(name, broadcastName); }
+    @Override public void disconnect() { close(); }
 
 }