Try to handle disconnecting