Fix recovery after dropped connection