websocket-server: Update for new ws-server