fix test bug