Improve WebSocket source component implementation, now working.