Fixing minor tree bug.