Fix server disconnect