Minor fix in test script.