diff --git a/rtcllm/rtc.py b/rtcllm/rtc.py
index b027c76..6203080 100644
--- a/rtcllm/rtc.py
+++ b/rtcllm/rtc.py
@@ -11,7 +11,7 @@ from functools import partial
 from appPublic.dictObject import DictObject
 from appPublic.hf import hf_socks5proxy
 from appPublic.worker import awaitify
-from aiortc import MediaStreamTrack, RTCPeerConnection, RTCSessionDescription, RTCIceCandidate
+from aiortc import MediaStreamTrack, VideoStreamTrack, AudioStreamTrack, RTCPeerConnection, RTCSessionDescription, RTCIceCandidate
 from aiortc.sdp import candidate_from_sdp, candidate_to_sdp
 from aiortc.contrib.media import MediaBlackhole, MediaPlayer, MediaRecorder, MediaRelay