Minor fix to expect script