Fixed the test script.