Fix error in test script