Fix bug when the leaf is not at the bottom

utreexo · Aug 26, 2024 · a4e4f6b · a4e4f6b
1 parent f3bd97b
commit a4e4f6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytreexo.py b/pytreexo.py
@@ -78,7 +78,7 @@ def root_idx(numleaves: int, position: int) -> int:
         if not root_present(numleaves, row):
             continue
         pos = position
-        for _ in range(row): pos = parent(pos, tree_rows(numleaves))
+        for _ in range(detect_row(position, tree_rows(numleaves)), row): pos = parent(pos, tree_rows(numleaves))
         if isroot(pos, numleaves, tree_rows(numleaves)):
             return idx
         idx += 1