[layouts][fix] use correct root reference in tree manager