diff --git a/src/openpi/training/config.py b/src/openpi/training/config.py
index a59cd1cf67..35166ae1e8 100644
--- a/src/openpi/training/config.py
+++ b/src/openpi/training/config.py
@@ -733,7 +733,7 @@ def __post_init__(self) -> None:
     ),
     TrainConfig(
         name="pi05_libero",
-        model=pi0_config.Pi0Config(pi05=True, action_horizon=10, discrete_state_input=False),
+        model=pi0_config.Pi0Config(pi05=True, action_horizon=10),
         data=LeRobotLiberoDataConfig(
             repo_id="physical-intelligence/libero",
             base_config=DataConfig(prompt_from_task=True),