update test script according to code changes