Fixed problem in test script