Reworking test script
Reworking test script