1 files changed, 2 insertions, 1 deletions
diff --git a/src/main.zig b/src/main.zig
index 42a4d34..a1414c8 100644
--- a/src/main.zig
+++ b/src/main.zig
@@ -1,4 +1,5 @@
 const std = @import("std");
+const builtin = @import("builtin");
 const zits = @import("zits");
 const yazap = @import("yazap");
 
@@ -8,7 +9,7 @@ const Server = zits.Server;
 pub fn main() !void {
     var dba: std.heap.DebugAllocator(.{}) = .init;
     defer _ = dba.deinit();
-    const gpa = dba.allocator();
+    const gpa = if (builtin.mode == .Debug or builtin.mode == .ReleaseSafe) dba.allocator() else std.heap.smp_allocator;
 
     var app = yazap.App.init(gpa, "zits", "High performance NATS compatible client and server.");
     defer app.deinit();