small tweak to tree