modify the test

modify the test