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 {