diff --git a/pyinfra/server/dispatcher/dispatchers/queue.py b/pyinfra/server/dispatcher/dispatchers/queue.py
index 390cc52..b61374b 100644
--- a/pyinfra/server/dispatcher/dispatchers/queue.py
+++ b/pyinfra/server/dispatcher/dispatchers/queue.py
@@ -1,13 +1,14 @@
 from pyinfra.server.dispatcher.dispatcher import Dispatcher
+from pyinfra.server.stream.queued_stream_function import QueuedStreamFunction
 
 
 class QueuedStreamFunctionDispatcher(Dispatcher):
-    def __init__(self, queue):
-        self.queue = queue
+    def __init__(self, queued_stream_function: QueuedStreamFunction):
+        self.queued_stream_function = queued_stream_function
 
     def patch(self, package):
-        self.queue.push(package)
-        return self.queue.pop()
+        self.queued_stream_function.push(package)
+        return self.queued_stream_function.pop()
 
     def post(self, package):
         return self.patch(package)