From 3ec9bfd2df433ec62a5638487b9df3f5e9ee0295 Mon Sep 17 00:00:00 2001
From: Hoon <48665813+sh-cha@users.noreply.github.com>
Date: Mon, 18 Nov 2024 23:41:54 +0900
Subject: [PATCH] change proof query error (#46)

---
 merkle/merkle.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/merkle/merkle.go b/merkle/merkle.go
index 66e368f..0f9f50c 100644
--- a/merkle/merkle.go
+++ b/merkle/merkle.go
@@ -310,8 +310,10 @@ func (m *Merkle) GetProofs(leafIndex uint64) (proofs [][]byte, treeIndex uint64,
 	}
 
 	// Check if the leaf index is in the tree
-	if leafIndex < treeInfo.StartLeafIndex || leafIndex-treeInfo.StartLeafIndex >= treeInfo.LeafCount {
+	if leafIndex < treeInfo.StartLeafIndex {
 		return nil, 0, nil, nil, fmt.Errorf("leaf (`%d`) is not found in tree (`%d`)", leafIndex, treeInfo.TreeIndex)
+	} else if leafIndex-treeInfo.StartLeafIndex >= treeInfo.LeafCount {
+		return nil, 0, nil, nil, fmt.Errorf("the tree containing the leaf (`%d`) has not been finalized yet", leafIndex)
 	}
 
 	height := uint8(0)