fix new unit test