diff --git a/tests/integration/functions/test_wal_api.rs b/tests/integration/functions/test_wal_api.rs
index e096e6729..198e9d9a1 100644
--- a/tests/integration/functions/test_wal_api.rs
+++ b/tests/integration/functions/test_wal_api.rs
@@ -41,10 +41,12 @@ fn test_wal_frame_transfer_no_schema_changes() {
     assert_eq!(conn1.wal_frame_count().unwrap(), 15);
     let mut frame = [0u8; 24 + 4096];
     conn2.wal_insert_begin().unwrap();
-    for frame_id in 1..=conn1.wal_frame_count().unwrap() as u32 {
+    let frames_count = conn1.wal_frame_count().unwrap() as u32;
+    for frame_id in 1..=frames_count {
         conn1.wal_get_frame(frame_id, &mut frame).unwrap();
         conn2.wal_insert_frame(frame_id, &frame).unwrap();
     }
+
     conn2.wal_insert_end().unwrap();
     assert_eq!(conn2.wal_frame_count().unwrap(), 15);
     assert_eq!(
@@ -134,12 +136,17 @@ fn test_wal_frame_transfer_schema_changes() {
         .unwrap();
     assert_eq!(conn1.wal_frame_count().unwrap(), 15);
     let mut frame = [0u8; 24 + 4096];
+    let mut commits = 0;
     conn2.wal_insert_begin().unwrap();
     for frame_id in 1..=conn1.wal_frame_count().unwrap() as u32 {
         conn1.wal_get_frame(frame_id, &mut frame).unwrap();
-        conn2.wal_insert_frame(frame_id, &frame).unwrap();
+        let info = conn2.wal_insert_frame(frame_id, &frame).unwrap();
+        if info.is_commit {
+            commits += 1;
+        }
     }
     conn2.wal_insert_end().unwrap();
+    assert_eq!(commits, 3);
     assert_eq!(conn2.wal_frame_count().unwrap(), 15);
     assert_eq!(
         limbo_exec_rows(&db2, &conn2, "SELECT x, length(y) FROM t"),