fix one unit test