diff --git a/R/treatment_effect.R b/R/treatment_effect.R
index 92f7415..d32d027 100644
--- a/R/treatment_effect.R
+++ b/R/treatment_effect.R
@@ -183,7 +183,7 @@ print.treatment_effect <- function(x, ...) {
   cat("Randomization: ", deparse(attr(x, "treatment")), "\n")
   cat("Marginal Mean: \n")
   print(attr(x, "marginal_mean"))
-  
+
   cat("Variance Type: ", attr(x, "vartype"), "\n")
   if (identical(attr(x, "vartype"), "none")) {
     trt_sd <- rep(NA, length(x))
@@ -207,5 +207,5 @@ print.treatment_effect <- function(x, ...) {
     coef_mat,
     zap.ind = 3,
     digits = 3
-  )  
+  )
 }
diff --git a/tests/testthat/test-utils.R b/tests/testthat/test-utils.R
index 6da6ffd..580c099 100644
--- a/tests/testthat/test-utils.R
+++ b/tests/testthat/test-utils.R
@@ -11,7 +11,7 @@ test_that("h_get_vars works for formula", {
     "only one randomization schema is allowed!"
   )
   expect_error(
-    h_get_vars( ~ ps(s2)),
+    h_get_vars(~ ps(s2)),
     "treatment formula must be of type treatment ~ strata"
   )
   expect_error(