diff --git a/src/simudator/core/modules/demux.py b/src/simudator/core/modules/demux.py
index b5f7bb7e477ad5215f7db74ab2194a69ff1ae999..3ab7d0e5660065e9b390962bb259460e3213ece4 100644
--- a/src/simudator/core/modules/demux.py
+++ b/src/simudator/core/modules/demux.py
@@ -92,8 +92,8 @@ class Demux(Module):
         Parameters
         ----------
         state : dict[str, Any]
-            The state of the demux to load. Should contain the keys "value",
-            "bit_length" and "mask", all with values of type ``int``.
+            The state of the demux to load. Should contain the keys "value" and
+            "bit_length", both with values of type ``int``.
         """
         super().set_state(state)
         self._value = state["value"]