fix unit test
fix unit test