Minor fix in client