more reference references correctly

epiforecasts · Nov 19, 2024 · 9d78c5e · 9d78c5e
1 parent 57dc93c
commit 9d78c5e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/R/as_contact_survey.R b/R/as_contact_survey.R
@@ -31,7 +31,7 @@ as_contact_survey <- function(x, id.column = "part_id",
   assert_names(names(x), must.include = c("participants", "contacts"))
   assert_data_frame(x$participants)
   assert_data_frame(x$contacts)
-  assert_list(reference, names = "named", null.ok = TRUE)
+  assert_list(x$reference, names = "named", null.ok = TRUE)
   assert_character(id.column)
   assert_character(year.column, null.ok = TRUE)
   assert_character(country.column, null.ok = TRUE)
@@ -59,11 +59,11 @@ as_contact_survey <- function(x, id.column = "part_id",
     }
   })
 
-  if (is.null(reference)) {
+  if (is.null(x$reference)) {
     warning("No reference provided")
   }
 
-  survey <- new_contact_survey(x$participant, x$contacts, reference)
+  survey <- new_contact_survey(x$participant, x$contacts, x$reference)
   survey <- clean(survey)
 
   return(survey)