fix test failure.