fix unit test one more time