Minor fix for disconnect hang.