diff --git a/zig/tests/cuda/contract_flashnext_fexp.zig b/zig/tests/cuda/contract_flashnext_fexp.zig
index 8389b4ba13..efe7a28257 100644
--- a/zig/tests/cuda/contract_flashnext_fexp.zig
+++ b/zig/tests/cuda/contract_flashnext_fexp.zig
@@ -79,7 +79,9 @@ pub const Fexp = struct {
         const a = self.rig.gpu.gpa;
         for (self.aff.items) |f| f.deinit();
         self.aff.clearRetainingCapacity();
-        for ([_][]const u8{ std.mem.sliceAsBytes(self.expert), std.mem.sliceAsBytes(self.gs), std.mem.sliceAsBytes(self.us) }) |b| if (b.len > 0) a.free(b);
+        a.free(self.expert);
+        a.free(self.gs);
+        a.free(self.us);
         self.expert = &.{};
         self.gs = &.{};
         self.us = &.{};
diff --git a/zig/tests/cuda/contract_glm_attn.zig b/zig/tests/cuda/contract_glm_attn.zig
index af7ed1de7c..b468b72ed5 100644
--- a/zig/tests/cuda/contract_glm_attn.zig
+++ b/zig/tests/cuda/contract_glm_attn.zig
@@ -195,7 +195,7 @@ pub const Chunks = struct {
             ds[i] = abs * scale * 520 * 0x1p-24;
             most = pages.maxNan(most, s[i]);
         }
-        for (p, s) |*x, v| x.* = @exp(v - most);
+        for (p, s[0..n]) |*x, v| x.* = @exp(v - most);
         const l = if (acc) |k| pages.sumIn(k, order, p) else pages.sumIn(.none, .sequential, p);
         const col = try a.alloc(f64, n);
         defer a.free(col);
diff --git a/zig/tests/cuda/contract_window_rows.zig b/zig/tests/cuda/contract_window_rows.zig
index cb0e1e9cac..2b36fce2fd 100644
--- a/zig/tests/cuda/contract_window_rows.zig
+++ b/zig/tests/cuda/contract_window_rows.zig
@@ -246,7 +246,7 @@ pub const Rows = struct {
         const a = self.arena.allocator();
         const f = self.f;
         const kv = f.hk * f.d;
-        for (0..kc.len / kv) |i| @memcpy(kc[i * kv ..][0..kv], kc[0..kv]);
+        for (1..kc.len / kv) |i| @memcpy(kc[i * kv ..][0..kv], kc[0..kv]); // the rows after the first take its keys
         @memset(vc, 0);
         const n = self.call.rows;
         const lo = self.call.first(self.call.at(n - 1).p);