Added test
Fixed test