We shouldn't need this class since we only need to set the onmessage callback