diff --git a/src/lerobot/annotations/steerable_pipeline/modules/advantage.py b/src/lerobot/annotations/steerable_pipeline/modules/advantage.py
index b010b1cc0..a3909c607 100644
--- a/src/lerobot/annotations/steerable_pipeline/modules/advantage.py
+++ b/src/lerobot/annotations/steerable_pipeline/modules/advantage.py
@@ -53,6 +53,7 @@ class AdvantageModule:
     """
 
     config: AdvantageConfig
+    frame_provider: Any = None
     _model: Any = field(default=None, init=False, repr=False)
     _preprocessor: Any = field(default=None, init=False, repr=False)
     _threshold: float | None = field(default=None, init=False, repr=False)