fix some error in unit test