diff --git a/pyinfra/queue/consumer.py b/pyinfra/queue/consumer.py
index a446b85..c21cb26 100644
--- a/pyinfra/queue/consumer.py
+++ b/pyinfra/queue/consumer.py
@@ -10,4 +10,4 @@ class Consumer:
         self.queue_manager.consume_and_publish(self.callback)
 
     def consume(self, **kwargs):
-        yield from self.queue_manager.consume(**kwargs)
+        return self.queue_manager.consume(**kwargs)