diff --git a/R/kmg01.R b/R/kmg01.R
index 408fdda9b..caa5536e9 100644
--- a/R/kmg01.R
+++ b/R/kmg01.R
@@ -38,7 +38,7 @@ kmg01_main <- function(adam_db,
   )
   assert_single_value(adam_db[[dataset]]$PARAMCD, label = paste0(df_lbl, "$PARAMCD"))
   assert_valid_variable(adam_db[[dataset]], "USUBJID", empty_ok = TRUE, types = list(c("character", "factor")))
-  variables <- list(tte = "AVAL", is_event = "IS_EVENT", arm = arm_var, strat = strat)
+  variables <- list(tte = "AVAL", is_event = "IS_EVENT", arm = arm_var, strata = strat)
   control_cox <- execute_with_args(control_coxph, ...)
   control_surv <- execute_with_args(control_surv_timepoint, ...)
   execute_with_args(
diff --git a/R/ttet01.R b/R/ttet01.R
index a2c36ce8b..46b83d244 100644
--- a/R/ttet01.R
+++ b/R/ttet01.R
@@ -142,7 +142,7 @@ ttet01_lyt <- function(arm_var,
         vars = "AVAL",
         is_event = "IS_EVENT",
         var_labels = if (perform == "strat") "Stratified Analysis" else "Unstratified Analysis",
-        strat = if (perform == "strat") strata else NULL,
+        strata = if (perform == "strat") strata else NULL,
         control = control_cox_ph,
         table_names = if (perform == "strat") "coxph_stratified" else "coxph_unstratified"
       )