diff --git a/src/autora/workflow/__init__.py b/src/autora/workflow/__init__.py
index 53957f63..9444603b 100644
--- a/src/autora/workflow/__init__.py
+++ b/src/autora/workflow/__init__.py
@@ -464,7 +464,7 @@
     Wen can run the seed pipeline with no data:
     >>> experimentalist_which_needs_no_data = make_pipeline([
     ...     np.linspace(*variables_2.independent_variables[0].value_range, 1_000),
-    ...     partial(random_sampler, n=10)]
+    ...     partial(random_sampler, num_samples=10)]
     ... )
     >>> np.array(experimentalist_which_needs_no_data())
     array([ 6.71671672, -0.73073073, -5.05505506,  6.13613614,  0.03003003,
diff --git a/tests/test_controller_plots.py b/tests/test_controller_plots.py
index 8d7b90df..2a5da2ea 100644
--- a/tests/test_controller_plots.py
+++ b/tests/test_controller_plots.py
@@ -52,7 +52,7 @@ def state_lr(ground_truth_1x):
         ],
         params={
             "pool": {"ivs": study_variables.independent_variables},
-            "sampler": {"n": 5},
+            "sampler": {"num_samples": 5},
         },
     )
 
@@ -117,7 +117,7 @@ def cycle_multi_lr(ground_truth_2x):
         ],
         params={
             "pool": {"ivs": study_variables.independent_variables},
-            "sampler": {"n": 10},
+            "sampler": {"num_samples": 10},
         },
     )