diff --git a/api/core/virtual_environment/channel/queue_transport.py b/api/core/virtual_environment/channel/queue_transport.py
index fa1114ca91..7cf524316a 100644
--- a/api/core/virtual_environment/channel/queue_transport.py
+++ b/api/core/virtual_environment/channel/queue_transport.py
@@ -1,5 +1,3 @@
-from queue import Empty, Queue
-
 from core.virtual_environment.channel.exec import TransportEOFError
 from core.virtual_environment.channel.transport import TransportReadCloser
 
@@ -29,6 +27,8 @@ class QueueTransportReadCloser(TransportReadCloser):
         A write handler that writes data to a queue.
         """
 
+        from queue import Queue
+
         def __init__(self, queue: Queue[bytes | None]) -> None:
             self.queue = queue
 
@@ -70,6 +70,7 @@ class QueueTransportReadCloser(TransportReadCloser):
 
         NEVER USE IT IN A MULTI-THREADED CONTEXT WITHOUT PROPER SYNCHRONIZATION.
         """
+        from queue import Empty
 
         if n <= 0:
             return b""