Fixes WebSocket endpoints - still not running properly