Minor fix for WS disconnect event