Fixed test error
Fixed test error