Update unit_tests.sh