Fix reconnection issue