modify test

modify test