diff --git a/tests/poseidon.rs b/tests/poseidon.rs
index 598f531..8426067 100644
--- a/tests/poseidon.rs
+++ b/tests/poseidon.rs
@@ -103,6 +103,14 @@ impl Database for MySled {
     }
 
     fn put_batch(&mut self, subtree: HashMap<DBKey, Value>) -> Result<()> {
+        let mut batch = sled::Batch::default();
+
+        for (key, value) in subtree {
+            batch.insert(&key, value);
+        }
+
+        self.0.apply_batch(batch).unwrap();
+
         Ok(())
     }
 }
diff --git a/tests/sled_keccak.rs b/tests/sled_keccak.rs
index 4f93af7..490a90e 100644
--- a/tests/sled_keccak.rs
+++ b/tests/sled_keccak.rs
@@ -41,6 +41,14 @@ impl Database for MySled {
     }
 
     fn put_batch(&mut self, subtree: HashMap<DBKey, Value>) -> Result<()> {
+        let mut batch = sled::Batch::default();
+
+        for (key, value) in subtree {
+            batch.insert(&key, value);
+        }
+
+        self.0.apply_batch(batch).unwrap();
+
         Ok(())
     }
 }