fix disconnection error