diff --git a/src/dvclive/live.py b/src/dvclive/live.py
index 7f3df0de..58924cf7 100644
--- a/src/dvclive/live.py
+++ b/src/dvclive/live.py
@@ -449,7 +449,7 @@ def cache(self, path):
         if self._inside_dvc_exp:
             msg = f"Skipping dvc add {path} because `dvc exp run` is running."
             path_stage = None
-            for stage in self._dvc_repo.stage.collect():
+            for stage in self._dvc_repo.index.stages:
                 for out in stage.outs:
                     if out.fspath == str(Path(path).absolute()):
                         path_stage = stage