Set current tree when ready