diff --git a/src/lerobot/policies/pi05/processor_pi05openpi.py b/src/lerobot/policies/pi05/processor_pi05openpi.py
index e06ae5027..545546e06 100644
--- a/src/lerobot/policies/pi05/processor_pi05openpi.py
+++ b/src/lerobot/policies/pi05/processor_pi05openpi.py
@@ -32,7 +32,7 @@ class Pi05PrepareStateTokenizerProcessorStep(ProcessorStep):
     Processor step to prepare the state and tokenize the language input.
     """
 
-    max_state_dim: int
+    max_state_dim: int = 32
     task_key: str = "task"
 
     def __call__(self, transition: EnvTransition) -> EnvTransition: