fix test

fix test