Could we add this to the prototype to make it easier to use from custom trees (and possible override)?