Fixing duty manage

MikeyUsersREC · Jan 21, 2024 · 62987bf · 62987bf
1 parent 2335020
commit 62987bf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/menus.py b/menus.py
@@ -6119,6 +6119,9 @@ async def on_duty_button(self, interaction: discord.Interaction, _: discord.Butt
                 color=blank_color
             ), ephemeral=True)
 
+        if self.state == 'on' or self.state == 'break':
+            return await self.cycle_ui(self.state, interaction.message)
+
 
         object_id = await self.bot.shift_management.add_shift_by_user(
             interaction.user,