Add TreeCapitator compatibility