From 9222310701e44ec73e157573aef985cb898c9f61 Mon Sep 17 00:00:00 2001
From: rymnc <43716372+rymnc@users.noreply.github.com>
Date: Mon, 24 Apr 2023 18:23:27 +0530
Subject: [PATCH] fix: tests

---
 src/tree.rs            | 2 +-
 tests/memory_keccak.rs | 2 +-
 tests/sled_keccak.rs   | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/tree.rs b/src/tree.rs
index d70b772..459c342 100644
--- a/src/tree.rs
+++ b/src/tree.rs
@@ -239,7 +239,7 @@ where
         Ok(())
     }
 
-    pub fn set_range(&mut self, start: usize, leaves: &[H::Fr]) -> Result<()> {
+    pub fn set_range(&mut self, start: usize, leaves: &[H::Fr]) -> PmtreeResult<()> {
         self.batch_insert(Some(start), leaves)
     }
 
diff --git a/tests/memory_keccak.rs b/tests/memory_keccak.rs
index 8971a38..4314be0 100644
--- a/tests/memory_keccak.rs
+++ b/tests/memory_keccak.rs
@@ -133,7 +133,7 @@ fn batch_insertions() -> PmtreeResult<()> {
 }
 
 #[test]
-fn set_range() -> Result<()> {
+fn set_range() -> PmtreeResult<()> {
     let mut mt = MerkleTree::<MemoryDB, MyKeccak>::new(2, MemoryDBConfig)?;
 
     let leaves = [
diff --git a/tests/sled_keccak.rs b/tests/sled_keccak.rs
index d5f346c..b9cdcca 100644
--- a/tests/sled_keccak.rs
+++ b/tests/sled_keccak.rs
@@ -172,7 +172,7 @@ fn batch_insertions() -> PmtreeResult<()> {
 }
 
 #[test]
-fn set_range() -> Result<()> {
+fn set_range() -> PmtreeResult<()> {
     let mut mt = MerkleTree::<MySled, MyKeccak>::new(
         2,
         SledConfig {