fix unit test scripts