1 files changed, 8 insertions, 0 deletions

diff --git a/src/FoldData.zig b/src/FoldData.zig
index e387447..a06eefe 100644
--- a/src/FoldData.zig
+++ b/src/FoldData.zig
@@ -24,6 +24,13 @@ pub fn init(allocator: mem.Allocator) !Self {
         .cwcf = try allocator.alloc(bool, 0x110000),
     };
 
+    var slices: usize = 0;
+    errdefer {
+        for (self.fold[0..slices]) |slice| self.allocator.free(slice);
+        self.allocator.free(self.fold);
+        self.allocator.free(self.cwcf);
+    }
+
     @memset(self.fold, &.{});
     @memset(self.cwcf, false);
 
@@ -32,6 +39,7 @@ pub fn init(allocator: mem.Allocator) !Self {
         if (len == 0) break;
         const cp = try reader.readInt(u24, endian);
         self.fold[cp >> 1] = try allocator.alloc(u21, len - 1);
+        slices += 1;
         for (0..len - 1) |i| {
             self.fold[cp >> 1][i] = @intCast(try reader.readInt(u24, endian));
         }