updated unit-test script