Fix WebSocket support (should now be complete)