diff --git a/src/lib/VncScreen.tsx b/src/lib/VncScreen.tsx
index 59222bd..a862d81 100644
--- a/src/lib/VncScreen.tsx
+++ b/src/lib/VncScreen.tsx
@@ -11,7 +11,7 @@ type EventListeners = { [T in NoVncEventType]?: (event: NoVncEvents[T]) => void
 
 export interface Props {
     url?: string;
-    websocket?: WebSocket;
+    websocket?: WebSocket | RTCDataChannel;
     style?: object;
     className?: string;
     viewOnly?: boolean;