fix websocket integration tests