fixed test script

fixed test script