refactor(imt.sol): use the cached tree depth

re privacy-scaling-explorations#209
aakansha1234 · Mar 20, 2024 · 4d5bf81 · 4d5bf81
1 parent 22b6a12
commit 4d5bf81
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/imt.sol/contracts/internal/InternalLeanIMT.sol b/packages/imt.sol/contracts/internal/InternalLeanIMT.sol
@@ -54,7 +54,7 @@ library InternalLeanIMT {
         // A new insertion can increase a tree's depth by at most 1,
         // and only if the number of leaves supported by the current
         // depth is less than the number of leaves to be supported after insertion.
-        if (2 ** self.depth < index + 1) {
+        if (2 ** treeDepth < index + 1) {
             ++treeDepth;
         }