1 files changed, 7 insertions, 6 deletions

diff --git a/src/main.zig b/src/main.zig
index a0b5288..2b7dca2 100644
--- a/src/main.zig
+++ b/src/main.zig
@@ -41,19 +41,19 @@ pub fn main() !void {
     };
     defer res.deinit();
 
-    try Saprus.init();
-    defer Saprus.deinit();
+    try SaprusClient.init();
+    defer SaprusClient.deinit();
 
     if (res.args.help != 0) {
         return clap.help(std.io.getStdErr().writer(), clap.Help, &params, .{});
     }
 
     if (res.args.relay) |r| {
-        try Saprus.sendRelay(if (r.len > 0) r else "Hello darkness my old friend", gpa);
+        try SaprusClient.sendRelay(if (r.len > 0) r else "Hello darkness my old friend", gpa);
         // std.debug.print("Sent: {s}\n", .{r});
         return;
     } else if (res.args.connect) |c| {
-        const conn_res: ?SaprusMessage = Saprus.connect(if (c.len > 0) c else "Hello darkness my old friend", gpa) catch |err| switch (err) {
+        const conn_res: ?SaprusMessage = SaprusClient.connect(if (c.len > 0) c else "Hello darkness my old friend", gpa) catch |err| switch (err) {
             error.WouldBlock => null,
             else => return err,
         };
@@ -74,7 +74,8 @@ const std = @import("std");
 const DebugAllocator = std.heap.DebugAllocator(.{});
 const ArrayList = std.ArrayList;
 
-const Saprus = @import("./saprus.zig");
-const SaprusMessage = Saprus.SaprusMessage;
+const zaprus = @import("zaprus");
+const SaprusClient = zaprus.Client;
+const SaprusMessage = zaprus.Message;
 
 const clap = @import("clap");