Adapt to new treebeard API.