Fixing the websockets setup.