Fix panic during disconnection