unit-test fix