fix test issue