Fix test
Fixed test