diff --git a/src/lerobot/configs/train.py b/src/lerobot/configs/train.py
index f56061e4e..546c4dbe0 100644
--- a/src/lerobot/configs/train.py
+++ b/src/lerobot/configs/train.py
@@ -63,8 +63,7 @@ class TrainPipelineConfig(HubMixin):
     scheduler: LRSchedulerConfig | None = None
     eval: EvalConfig = field(default_factory=EvalConfig)
     wandb: WandBConfig = field(default_factory=WandBConfig)
-    use_peft: bool = False
-    peft: PeftConfig = field(default_factory=PeftConfig)
+    peft: PeftConfig | None = None
 
     def __post_init__(self):
         self.checkpoint_path = None
diff --git a/src/lerobot/scripts/lerobot_train.py b/src/lerobot/scripts/lerobot_train.py
index c608c855f..4bc197f4d 100644
--- a/src/lerobot/scripts/lerobot_train.py
+++ b/src/lerobot/scripts/lerobot_train.py
@@ -247,7 +247,7 @@ def train(cfg: TrainPipelineConfig):
         ds_meta=dataset.meta,
     )
 
-    if cfg.use_peft:
+    if cfg.peft is not None:
         logging.info("Using PEFT! Wrapping model.")
         policy = wrap_policy_in_peft_model(cfg, policy)