diff --git a/test/browser/highlight_block_callbacks.js b/test/browser/highlight_block_callbacks.js
index fdde5d526e..d42ae39f79 100644
--- a/test/browser/highlight_block_callbacks.js
+++ b/test/browser/highlight_block_callbacks.js
@@ -141,7 +141,7 @@ describe('after:highlightElement', function() {
     });
 
     this.hljs.highlightElement(this.block);
-    this.block.outerHTML.should.equal(`<code class="language-javascript hljs">redacted</code>`);
+    this.block.outerHTML.should.equal(`<code class="language-javascript hljs" data-highlighted="yes">redacted</code>`);
   })
 })