diff --git a/analysis/km.R b/analysis/km.R
index 61f6f4e..220af39 100644
--- a/analysis/km.R
+++ b/analysis/km.R
@@ -205,8 +205,8 @@ if(length(exposure)>0){
       max_fup_time = max(event_time),
       max_event_time = max(event_time[event_indicator])
     )
-  cat("maximum follow-up time is [", paste0(max_time_data$max_fup_time, collapse= ", "), "]", "\\n")
-  cat("maximum event time is [", paste0(max_time_data$max_event_time, collapse= ", "), "]", "\\n")
+  cat("maximum follow-up time is [", paste0(max_time_data$max_fup_time, collapse= ", "), "]", "\n")
+  cat("maximum event time is [", paste0(max_time_data$max_event_time, collapse= ", "), "]", "\n")
 }
 
 
@@ -470,9 +470,18 @@ if(smooth){
 km_plot <- function(.data) {
 
   data_with_time0 <-
+  if(length(exposure)>0L){
     .data |>
+      mutate(
+        "{exposure}" := as.factor(!!!exposure_syms)
+      )
+  } else {
+    .data
+  }
+
+  data_with_time0 <-
+    data_with_time0 |>
     mutate(
-      "{exposure}" := as.factor(!!!exposure_syms),
       lagtime = lag(time, 1, 0), # assumes the time-origin is zero
     ) %>%
     group_modify(
@@ -486,6 +495,7 @@ km_plot <- function(.data) {
         .before = 0
       )
     )
+
   ggplot_init <- if(length(exposure)==0L){
     ggplot(data_with_time0)
   } else {