fix unit test

fix unit test