diff --git a/menus.py b/menus.py
index 77dda16..d578c30 100644
--- 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,