diff --git a/flanabot/bots/flana_bot.py b/flanabot/bots/flana_bot.py index 73a0c07..d8fddf1 100644 --- a/flanabot/bots/flana_bot.py +++ b/flanabot/bots/flana_bot.py @@ -422,9 +422,14 @@ class FlanaBot(Connect4Bot, PenaltyBot, PollBot, ScraperBot, UberEatsBot, Weathe and message.replied_message.medias ): - whitelist_callbacks = (whitelist_callbacks or set()) | {self._on_song_info, self._on_delete} + whitelist_callbacks = (whitelist_callbacks or set()) | { + self._on_delete, + self._on_recover_message, + self._on_reset, + self._on_song_info + } elif self.tunnel_chat and message.chat == await self.owner_chat: - whitelist_callbacks = (whitelist_callbacks or set()) | {self._on_tunnel_message, self._on_deactivate_tunnel} + whitelist_callbacks = (whitelist_callbacks or set()) | {self._on_deactivate_tunnel, self._on_tunnel_message} await super()._on_new_message_raw(message, whitelist_callbacks, blacklist_callbacks)