Modified unit test to reflect the right behaviour