Fix error on TreeBuilder.