Fix issue when modifying websocket example

Thanks ale.