From 8901c09eb85a8a123821b3de7071d9f7782e7e94 Mon Sep 17 00:00:00 2001
From: Emily de la Rua <emily.de_la_rua@contractors.roche.com>
Date: Fri, 22 Sep 2023 10:39:16 -0400
Subject: [PATCH] Minor refactor

---
 R/arm_ref_comp.R | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/R/arm_ref_comp.R b/R/arm_ref_comp.R
index cfbe128b9b..d80da412a7 100644
--- 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)
       }