diff --git a/tests/testthat/test-csv.R b/tests/testthat/test-csv.R
index 6dc331fe..6c3c31f0 100644
--- a/tests/testthat/test-csv.R
+++ b/tests/testthat/test-csv.R
@@ -6,7 +6,7 @@ fit_bernoulli_variational <- testing_fit("bernoulli", method = "variational", se
 fit_bernoulli_laplace <- testing_fit("bernoulli", method = "laplace", seed = 123)
 fit_logistic_optimize <- testing_fit("logistic", method = "optimize", seed = 123)
 fit_logistic_variational <- testing_fit("logistic", method = "variational", seed = 123)
-fit_logistic_variational_short <- testing_fit("logistic", method = "variational", output_samples = 100, seed = 123)
+fit_logistic_variational_short <- testing_fit("logistic", method = "variational", draws = 100, seed = 123)
 fit_logistic_laplace <- testing_fit("logistic", method = "laplace", seed = 123)
 
 fit_bernoulli_diag_e_no_samples <- testing_fit("bernoulli", method = "sample",
diff --git a/tests/testthat/test-fit-shared.R b/tests/testthat/test-fit-shared.R
index 0f98e64a..2b07abb8 100644
--- a/tests/testthat/test-fit-shared.R
+++ b/tests/testthat/test-fit-shared.R
@@ -437,7 +437,7 @@ test_that("draws are returned for model with spaces", {
   )
   expect_equal(dim(fit_sample$draws()), c(1000, 1, 3))
   utils::capture.output(
-    fit <- mod$variational(seed = 123, output_samples = 1000)
+    fit <- mod$variational(seed = 123, draws = 1000)
   )
   expect_equal(dim(fit$draws()), c(1000, 4))
   utils::capture.output(
diff --git a/tests/testthat/test-fit-vb.R b/tests/testthat/test-fit-vb.R
index 0e8a4dd9..e701e951 100644
--- a/tests/testthat/test-fit-vb.R
+++ b/tests/testthat/test-fit-vb.R
@@ -90,7 +90,7 @@ test_that("time is reported after vb", {
                     seed = 123,
                     elbo_samples = 1000,
                     iter = 2000,
-                    output_samples = 50
+                    draws = 50
                     ),
     "Finished in"
   )