1 files changed, 6 insertions, 2 deletions
diff --git a/src/Client.zig b/src/Client.zig
index 19f8360..c518c74 100644
--- a/src/Client.zig
+++ b/src/Client.zig
@@ -76,7 +76,7 @@ pub fn sendInitialConnection(payload: []const u8, initial_port: u16, allocator:
     return msg;
 }
 
-pub fn connect(payload: []const u8, allocator: Allocator) !?SaprusMessage {
+pub fn connect(payload: []const u8, allocator: Allocator) !?SaprusConnection {
     var initial_port: u16 = 0;
     if (rand) |r| {
         initial_port = r.intRangeAtMost(u16, 1024, 65000);
@@ -109,10 +109,14 @@ pub fn connect(payload: []const u8, allocator: Allocator) !?SaprusMessage {
     // Complete handshake after awaiting response
     try broadcastSaprusMessage(msg, randomPort(), allocator);
 
-    return initial_conn_res;
+    if (false) {
+        return initial_conn_res.?;
+    }
+    return null;
 }
 
 const SaprusMessage = @import("message.zig").Message;
+const SaprusConnection = @import("Connection.zig");
 
 const std = @import("std");
 const Random = std.Random;