Update unit test script