Update test script
Update test script