Fixed test to have correct behaviour.