diff --git a/R/reprex-undo.R b/R/reprex-undo.R
index d275dbd8..cf4a961b 100644
--- a/R/reprex-undo.R
+++ b/R/reprex-undo.R
@@ -232,9 +232,5 @@ classify_lines <- function(x, comment = "^#>") {
   wut <- ifelse(wut == "code" & grepl(comment, x), "output", wut)
 
   so_special <- "<!-- language-all: lang-r -->"
-  if (identical(x[1], so_special)) {
-    wut[1] <- "so_header"
-  }
-
-  wut
+  ifelse(x == so_special, "so_header", wut)
 }