fix test bug
fix test bug