diff --git a/river/evaluate/progressive_validation.py b/river/evaluate/progressive_validation.py
index 45e9e0d4f8..0fbb63884a 100644
--- a/river/evaluate/progressive_validation.py
+++ b/river/evaluate/progressive_validation.py
@@ -190,6 +190,8 @@ def iter_progressive_val_score(
 
     The `yield_predictions` parameter can be used to include the predictions in the results:
 
+    >>> import itertools
+
     >>> steps = evaluate.iter_progressive_val_score(
     ...     model=model,
     ...     dataset=datasets.Phishing(),