Minor clean up, improve missing web socket message