diff --git a/pybalboa/control.py b/pybalboa/control.py
index 8e12e15..c2f2677 100644
--- a/pybalboa/control.py
+++ b/pybalboa/control.py
@@ -188,7 +188,7 @@ async def set_state(self, state: int | HeatMode) -> bool:
             return False
         if self._state == state:
             return True
-        i = 2 if self.state == HeatMode.READY_IN_REST and state == HeatMode.READY else 1  # type: ignore
+        i = 2 if self.state == HeatMode.READY_IN_REST and state == HeatMode.READY else 1
         for _ in range(i):
             await self._client.send_message(MessageType.TOGGLE_STATE, self._code)
         return True