Updated to match the new api of web socket helper