fix ws connection