Fixed small bug with refactoring of Tree module.