diff --git a/survival.R b/survival.R
index e55a243..5bc657f 100644
--- a/survival.R
+++ b/survival.R
@@ -6,7 +6,7 @@
 library(survival)
 source("utils.R")
 
-secmal <- read.csv2("current.csv", header=TRUE)
+secmal <- sma_load_data("current.csv")
 
 # Setup survival object
 surv_dx <- Surv(time = secmal$event_time_dx, event = secmal$event_status)