Small Fix in test script

Small Fix in test script