1 files changed, 9 insertions, 4 deletions

diff --git a/spec/lib/banzai/pipeline/plain_markdown_pipeline_spec.rb b/spec/lib/banzai/pipeline/plain_markdown_pipeline_spec.rb
index 0e4a4e4492e..e7c15ed9cf6 100644
--- a/spec/lib/banzai/pipeline/plain_markdown_pipeline_spec.rb
+++ b/spec/lib/banzai/pipeline/plain_markdown_pipeline_spec.rb
@@ -15,10 +15,15 @@ RSpec.describe Banzai::Pipeline::PlainMarkdownPipeline, feature_category: :team_
       result = described_class.call(markdown, project: project)
       output = result[:output].to_html
 
-      Banzai::Filter::MarkdownPreEscapeFilter::ESCAPABLE_CHARS.pluck(:char).each do |char|
-        char = '&' if char == '&'
-
-        expect(output).to include("<span>#{char}</span>")
+      Banzai::Filter::MarkdownPreEscapeFilter::ESCAPABLE_CHARS.each do |item|
+        char = item[:char] == '&' ? '&amp;' : item[:char]
+
+        if item[:reference]
+          expect(output).to include("<span data-escaped-char>#{char}</span>")
+        else
+          expect(output).not_to include("<span data-escaped-char>#{char}</span>")
+          expect(output).to include(char)
+        end
       end
 
       expect(result[:escaped_literals]).to be_truthy