fix main test script