Fixed reconnect issue