fix test for new version