diff --git a/src/idom/core/types.py b/src/idom/core/types.py
index db366fcd5..0ef45c263 100644
--- a/src/idom/core/types.py
+++ b/src/idom/core/types.py
@@ -28,11 +28,11 @@
 
 if TYPE_CHECKING or sys.version_info < (3, 9) or sys.version_info >= (3, 11):
 
-    class State(NamedTuple, Generic[_Type]):  # pragma: no cover
+    class State(NamedTuple, Generic[_Type]):
         value: _Type
         set_value: Callable[[_Type | Callable[[_Type], _Type]], None]
 
-else:
+else:  # pragma: no cover
     State = namedtuple("State", ("value", "set_value"))