Update unit-test.sh