diff --git a/helpers/pygments_test.go b/helpers/pygments_test.go
index 830b3194428..e8a5199d9e5 100644
--- a/helpers/pygments_test.go
+++ b/helpers/pygments_test.go
@@ -141,7 +141,7 @@ func TestChromaHTMLHighlight(t *testing.T) {
 	result, err := spec.Highlight(`echo "Hello"`, "bash", "")
 	assert.NoError(err)
 
-	assert.Contains(result, `<code class="language-bash" data-lang="bash"><span class="nb">echo</span><span class=""> </span><span class="s2">&#34;Hello&#34;</span></code>`)
+	assert.Contains(result, `<div class="highlight"><pre class="chroma"><code class="language-bash" data-lang="bash"><span class="nb">echo</span> <span class="s2">&#34;Hello&#34;</span></code></pre></div>`)
 
 }