Fixed test

All tests pass