From c903cbf193f1a0436e58b30bb358da4b1edcf206 Mon Sep 17 00:00:00 2001
From: Vaci Koblizek <vaci@vaci.org>
Date: Sat, 24 Oct 2020 16:43:53 +0100
Subject: [PATCH] return this network via asConnection()

---
 runtime/src/main/java/org/capnproto/TwoPartyVatNetwork.java | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/runtime/src/main/java/org/capnproto/TwoPartyVatNetwork.java b/runtime/src/main/java/org/capnproto/TwoPartyVatNetwork.java
index d9f67d8..6196ade 100644
--- a/runtime/src/main/java/org/capnproto/TwoPartyVatNetwork.java
+++ b/runtime/src/main/java/org/capnproto/TwoPartyVatNetwork.java
@@ -46,13 +46,15 @@ public class TwoPartyVatNetwork
     }
 
     private Connection connect(RpcTwoPartyProtocol.VatId.Reader vatId) {
-        return vatId.getSide() != side ? this : null;
+        return vatId.getSide() != side
+                ? this.asConnection()
+                : null;
     }
 
     private CompletableFuture<Connection> accept() {
         if (side == RpcTwoPartyProtocol.Side.SERVER & !accepted) {
             accepted = true;
-            return CompletableFuture.completedFuture(this);
+            return CompletableFuture.completedFuture(this.asConnection());
         }
         else {
             // never completes