Fixed new unit test