Changed unit test