edited unit test