Fixed test break.
fixed test break