Fixed tree shrinking bug.