diff --git a/publish_subscribe.py b/publish_subscribe.py index 6c03c392..c748c1a4 100644 --- a/publish_subscribe.py +++ b/publish_subscribe.py @@ -24,9 +24,8 @@ def unsubscribe(self, msg, subscriber): def update(self): for msg in self.msg_queue: - if msg in self.subscribers: - for sub in self.subscribers[msg]: - sub.run(msg) + for sub in self.subscribers.get(msg, []): + sub.run(msg) self.msg_queue = []