Fixed reconnection issue