Minor update to test.sh