Attempt at fixing current WebSocket tests

r=@tOkeshu