fix one more unit test