diff --git a/R/likelihood.R b/R/likelihood.R
index f9c46bc9..4b103805 100644
--- a/R/likelihood.R
+++ b/R/likelihood.R
@@ -49,7 +49,7 @@
 #' # Example using an <epichains_tree> object
 #' set.seed(32)
 #' chains_tree_eg <- simulate_chains(
-#'  index_cases = 10,
+#'  n_chains = 10,
 #'  pop = 100,
 #'  percent_immune = 0,
 #'  statistic = "size",
@@ -73,7 +73,7 @@
 #' # Example using a <epichains_summary> object
 #' set.seed(32)
 #' chains_summary_eg <- simulate_chain_stats(
-#'  index_cases = 10,
+#'  n_chains = 10,
 #'   pop = 100,
 #'   percent_immune = 0,
 #'   statistic = "size",
diff --git a/man/likelihood.Rd b/man/likelihood.Rd
index 5dc099b1..4579d957 100644
--- a/man/likelihood.Rd
+++ b/man/likelihood.Rd
@@ -100,7 +100,7 @@ likelihood(
 # Example using an <epichains_tree> object
 set.seed(32)
 chains_tree_eg <- simulate_chains(
- index_cases = 10,
+ n_chains = 10,
  pop = 100,
  percent_immune = 0,
  statistic = "size",
@@ -124,7 +124,7 @@ chains_tree_eg_lik
 # Example using a <epichains_summary> object
 set.seed(32)
 chains_summary_eg <- simulate_chain_stats(
- index_cases = 10,
+ n_chains = 10,
   pop = 100,
   percent_immune = 0,
   statistic = "size",
diff --git a/tests/testthat/test-likelihood.R b/tests/testthat/test-likelihood.R
index 28e63e20..05f29bb2 100644
--- a/tests/testthat/test-likelihood.R
+++ b/tests/testthat/test-likelihood.R
@@ -93,7 +93,7 @@ test_that("likelihood() works with epichains and epichains_summary objects", {
     # Simulate an <epichains_tree> object
     set.seed(32)
     chains_tree_eg <- simulate_chains(
-      index_cases = 10,
+      n_chains = 10,
       pop = 100,
       percent_immune = 0,
       statistic = "size",
@@ -105,7 +105,7 @@ test_that("likelihood() works with epichains and epichains_summary objects", {
     # Simulate an <epichains_summary> object
     set.seed(32)
     chains_summary_eg <- simulate_chain_stats(
-      index_cases = 10,
+      n_chains = 10,
       pop = 100,
       percent_immune = 0,
       statistic = "size",