diff --git a/src/dailyai/transports/daily_transport.py b/src/dailyai/transports/daily_transport.py
index 2eab8759a..1da114d31 100644
--- a/src/dailyai/transports/daily_transport.py
+++ b/src/dailyai/transports/daily_transport.py
@@ -24,7 +24,7 @@
     VirtualSpeakerDevice,
 )
 
-from dailyai.transports.base_transport import ThreadedTransport
+from dailyai.transports.threaded_transport import ThreadedTransport
 
 
 class DailyTransport(ThreadedTransport, EventHandler):
@@ -48,7 +48,7 @@ def __init__(
         start_transcription: bool = False,
         **kwargs,
     ):
-        # This will call BaseTransportService.__init__ method, not EventHandler
+        # This will call ThreadedTransport.__init__ method, not EventHandler
         super().__init__(**kwargs)
 
         self._room_url: str = room_url
diff --git a/src/dailyai/transports/local_transport.py b/src/dailyai/transports/local_transport.py
index 4c330e024..1e791a467 100644
--- a/src/dailyai/transports/local_transport.py
+++ b/src/dailyai/transports/local_transport.py
@@ -3,7 +3,7 @@
 import tkinter as tk
 import pyaudio
 
-from dailyai.transports.base_transport import ThreadedTransport
+from dailyai.transports.threaded_transport import ThreadedTransport
 
 
 class LocalTransport(ThreadedTransport):
diff --git a/src/dailyai/transports/websocket_transport.py b/src/dailyai/transports/websocket_transport.py
index fc42a4cb9..84c185a3d 100644
--- a/src/dailyai/transports/websocket_transport.py
+++ b/src/dailyai/transports/websocket_transport.py
@@ -7,7 +7,7 @@
 from dailyai.pipeline.frames import AudioFrame, ControlFrame, EndFrame, Frame, TTSEndFrame, TTSStartFrame, TextFrame
 from dailyai.pipeline.pipeline import Pipeline
 from dailyai.serializers.protobuf_serializer import ProtobufFrameSerializer
-from dailyai.transports.base_transport import ThreadedTransport
+from dailyai.transports.threaded_transport import ThreadedTransport
 
 
 class WebSocketFrameProcessor(FrameProcessor):