Fixed tests
Fixed test