1 files changed, 4 insertions, 3 deletions
diff --git a/src/server/main.zig b/src/server/main.zig
index 7e3f503..0045c1c 100644
--- a/src/server/main.zig
+++ b/src/server/main.zig
@@ -100,7 +100,8 @@ fn handleConnection(
     var connect_arena: std.heap.ArenaAllocator = .init(allocator);
     defer connect_arena.deinit();
     const connect = (Message.next(connect_arena.allocator(), in) catch return).connect;
-    var client_state: ClientState = try .init(io, allocator, id, connect, in, out);
+    var client_state: ClientState = try .init(connect, in, out);
+    try client_state.start(io);
     defer client_state.deinit(io, allocator);
 
     try server.addClient(allocator, id, &client_state);
@@ -123,10 +124,10 @@ fn handleConnection(
                 }
             },
             .sub => |sub| {
-                try server.subscribe(allocator, client_state.id, sub);
+                try server.subscribe(allocator, id, sub);
             },
             .unsub => |unsub| {
-                try server.unsubscribe(client_state.id, unsub);
+                try server.unsubscribe(id, unsub);
             },
             else => |e| {
                 std.debug.panic("Unimplemented message: {any}\n", .{e});