Added a proper reconnect