diff --git a/tests/testthat/test-seasonal_burden_levels.R b/tests/testthat/test-seasonal_burden_levels.R
index 719e456..d7de5f1 100644
--- a/tests/testthat/test-seasonal_burden_levels.R
+++ b/tests/testthat/test-seasonal_burden_levels.R
@@ -52,11 +52,12 @@ test_that("Test that input argument checks work", {
 
   #Finally check that dots arguments work
   model_output <- seasonal_burden_levels(
-    tsd_data, family = "exp",
+    tsd_data,
+    family = "exp",
     optim_method = "Brent",
     lower_optim = 1, upper_optim = 1000
   )
-  expect_equal(model_output$family, "exp")
+  expect_equal(model_output$optim$family, "exp")
 })
 
 test_that("Test that we get correct season output for newest season", {