1 files changed, 3 insertions, 2 deletions
diff --git a/src/layout.zig b/src/layout.zig
index 116a64b8..3d8932a8 100644
--- a/src/layout.zig
+++ b/src/layout.zig
@@ -1,5 +1,6 @@
 const std = @import("std");
 const pardes = @import("pardes.zig");
+const dump = @import("dump.zig");
 const config = @import("config.zig");
 const panes = @import("panes.zig");
 const Pardes = pardes.Pardes;
@@ -1442,8 +1443,8 @@ test "collapse dump restores folded state and retained expansion weight" {
     compute(p);
     const original_height = p.rects[0].h;
     toggleCollapse(p, 0);
-    try p.dumpState();
-    const restored = try Pardes.initFromDump(std.testing.allocator, .{ .tty_only = true, .cols = 80, .rows = 32 }, p.dump_out.?);
+    try dump.dumpState(p);
+    const restored = try dump.initFromDump(std.testing.allocator, .{ .tty_only = true, .cols = 80, .rows = 32 }, p.dump_out.?);
     defer restored.deinit();
     try std.testing.expect(restored.panes[0].?.collapsed);
     try std.testing.expectEqual(@as(f32, @floatFromInt(original_height)), restored.panes[0].?.vweight);