Minor fix to test script