Fixed test after tree change.