diff --git a/src/components/MarkdownUi.spec.ts b/src/components/MarkdownUi.spec.ts
index bc9b65fc..a7868078 100644
--- a/src/components/MarkdownUi.spec.ts
+++ b/src/components/MarkdownUi.spec.ts
@@ -391,6 +391,7 @@ describe('<MarkdownUi />', () => {
       expect(wrapper.findTestId('markdown-content').classes('html-preview')).toBe(false)
       // Ensure the HTML preview does not exist
       expect(wrapper.find('pre > code.language-html').exists()).toBe(false)
+      expect(wrapper.findTestId('copy-code-button').exists()).toBe(false)
 
       // Both panes should be visible
       expect(wrapper.findTestId('markdown-editor-textarea').isVisible()).toBe(true)
@@ -412,6 +413,7 @@ describe('<MarkdownUi />', () => {
       expect(wrapper.findTestId('markdown-content').classes('html-preview')).toBe(true)
 
       expect(wrapper.find('pre > code.language-html').isVisible()).toBe(true)
+      expect(wrapper.findTestId('copy-code-button').isVisible()).toBe(true)
     })
   })
 })