diff --git a/src/lerobot/rollout/__init__.py b/src/lerobot/rollout/__init__.py
index 8ef6c26cd..da8c18acc 100644
--- a/src/lerobot/rollout/__init__.py
+++ b/src/lerobot/rollout/__init__.py
@@ -56,7 +56,6 @@ __all__ = [
     "DAggerPedalConfig",
     "DAggerStrategyConfig",
     "DatasetContext",
-    "DatasetRecordConfig",
     "HardwareContext",
     "HighlightStrategyConfig",
     "InferenceEngine",
diff --git a/tests/test_rollout.py b/tests/test_rollout.py
index 0a53ea477..dd3fb25b6 100644
--- a/tests/test_rollout.py
+++ b/tests/test_rollout.py
@@ -80,7 +80,7 @@ def test_dagger_config_defaults():
     from lerobot.rollout import DAggerStrategyConfig
 
     cfg = DAggerStrategyConfig()
-    assert cfg.num_episodes == 10
+    assert cfg.num_episodes is None
     assert cfg.record_autonomous is False
     assert cfg.input_device == "keyboard"