Minor fix, return websocket server object.