Fix test script
Fix test script