Minor refactor

insightsengineering · Sep 22, 2023 · 8901c09 · 8901c09
1 parent 3cf1df3
commit 8901c09
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/arm_ref_comp.R b/R/arm_ref_comp.R
@@ -108,7 +108,7 @@ arm_ref_comp_observer <- function(session,
       teal::validate_has_elements(arm, "Treatment variable is empty.")
 
       arm_levels <- if (is.factor(arm)) {
-        intersect(levels(arm), unique(arm))
+        levels(droplevels(arm))
       } else {
         unique(arm)
       }