diff --git a/inst/templates/unit_tests/tests/testthat/test-main.R b/inst/templates/unit_tests/tests/testthat/test-main.R
index 003ae7e8..b8ddd76a 100644
--- a/inst/templates/unit_tests/tests/testthat/test-main.R
+++ b/inst/templates/unit_tests/tests/testthat/test-main.R
@@ -3,7 +3,7 @@ box::use(
   testthat[expect_true, test_that],
 )
 box::use(
-  app/main[server, ui],
+  app/main[server],
 )
 
 test_that("main server works", {
diff --git a/tests/e2e/app-files/hello.R b/tests/e2e/app-files/hello.R
index 44f36947..2e85bf30 100644
--- a/tests/e2e/app-files/hello.R
+++ b/tests/e2e/app-files/hello.R
@@ -1,5 +1,5 @@
 box::use(
-  shiny[           # nolint
+  shiny[
     actionButton,
     bootstrapPage,
     isolate,
diff --git a/tests/e2e/app-files/say_hello.R b/tests/e2e/app-files/say_hello.R
index c69dbd17..7e221c9c 100644
--- a/tests/e2e/app-files/say_hello.R
+++ b/tests/e2e/app-files/say_hello.R
@@ -1,3 +1,4 @@
+#' @export
 say_hello <- function(name) {
   paste0("Hello, ", name, "!")
 }