Get rid of BTree.root_id and change BTree.root into a real attribute.

1 files changed, 2 insertions, 2 deletions

diff --git a/insert-remove-test b/insert-remove-test
index 19a5e86..08e7fd9 100755
--- a/insert-remove-test
+++ b/insert-remove-test
@@ -55,12 +55,12 @@ def assert_refcounts_are_one(tree):
         if isinstance(node, btree.IndexNode):
             for key, child_id in node.pairs():
                 helper(child_id)
-    helper(tree.root_id)
+    helper(tree.root.id)
 
 
 def do_insert(tree, key, value):
     logging.debug('do_insert(%s)' % (repr(key)))
-    if tree.root_id is not None:
+    if tree.root is not None:
         assert_refcounts_are_one(tree)
     tree.insert(key, value)
     assert_refcounts_are_one(tree)