diff --git a/tests/testthat/test-missing_data.R b/tests/testthat/test-missing_data.R
index 066243f..9e34190 100644
--- a/tests/testthat/test-missing_data.R
+++ b/tests/testthat/test-missing_data.R
@@ -2,9 +2,19 @@ data("breastcancer", package = "risks")
 df <- breastcancer |>
   dplyr::mutate(
     continuous = 1:dplyr::n(),
-    receptor = dplyr::if_else(dplyr::row_number() %in% 9:11, NA, receptor),
+    receptor = {
+      label <- attr(receptor, "label")
+      receptor <- dplyr::if_else(dplyr::row_number() %in% 9:11, NA, receptor)
+      attr(receptor, "label") <- label
+      receptor
+    },
     stage = dplyr::if_else(dplyr::row_number() %in% 29:31, NA, stage),
-    death = dplyr::if_else(dplyr::row_number() %in% 99:101, NA, death),
+    death = {
+      label <- attr(death, "label")
+      death <- dplyr::if_else(dplyr::row_number() %in% 99:101, NA, death)
+      attr(death, "label") <- label
+      death
+    },
     allempty = NA_real_,
     allempty_lgl = allempty == 1
   )
diff --git a/tests/testthat/test-table1_design.R b/tests/testthat/test-table1_design.R
index c3308cc..68f6ff1 100644
--- a/tests/testthat/test-table1_design.R
+++ b/tests/testthat/test-table1_design.R
@@ -2,8 +2,18 @@ data("breastcancer", package = "risks")
 df <- breastcancer |>
   dplyr::mutate(
     continuous = 1:dplyr::n(),
-    receptor = dplyr::if_else(dplyr::row_number() %in% 9:11, NA, receptor),
-    death = dplyr::if_else(dplyr::row_number() %in% 99:101, NA, death),
+    receptor = {
+      label <- attr(receptor, "label")
+      receptor <- dplyr::if_else(dplyr::row_number() %in% 9:11, NA, receptor)
+      attr(receptor, "label") <- label
+      receptor
+    },
+    death = {
+      label <- attr(death, "label")
+      death <- dplyr::if_else(dplyr::row_number() %in% 99:101, NA, death)
+      attr(death, "label") <- label
+      death
+    },
     allempty = NA_integer_,
     allempty_lgl = allempty == 1
   )
diff --git a/vignettes/table1.Rmd b/vignettes/table1.Rmd
index 9ac6543..2dadc36 100644
--- a/vignettes/table1.Rmd
+++ b/vignettes/table1.Rmd
@@ -16,7 +16,7 @@ knitr::opts_chunk$set(
 
 # Loading the package
 
-After installation (using `remotes::install_github("stopsack/rifttable")`), load the package with:
+After installation, load the package with:
 
 ```{r setup}
 library(rifttable)