diff --git a/src/lerobot/record.py b/src/lerobot/record.py
index 7a607d3c1..9cfbcad2b 100644
--- a/src/lerobot/record.py
+++ b/src/lerobot/record.py
@@ -305,7 +305,7 @@ def record(cfg: RecordConfig) -> LeRobotDataset:
 
     # Load pretrained policy
 
-    if cfg.policy.use_peft:
+    if cfg.policy and cfg.policy.use_peft:
         from peft import PeftModel
 
         logging.info("Loading policy's PEFT adapter.")
diff --git a/tests/utils/test_train_utils.py b/tests/utils/test_train_utils.py
index 1fc02db82..312b1398c 100644
--- a/tests/utils/test_train_utils.py
+++ b/tests/utils/test_train_utils.py
@@ -79,7 +79,7 @@ def test_save_checkpoint(mock_save_training_state, tmp_path, optimizer):
     mock_save_training_state.assert_called_once()
 
 
-@patch("lerobot.common.utils.train_utils.save_training_state")
+@patch("lerobot.utils.train_utils.save_training_state")
 def test_save_checkpoint_peft(mock_save_training_state, tmp_path, optimizer):
     policy = Mock()
     policy.config = Mock()