Fix WS reconnection