Modify test

Modify Test