diff --git a/packages/astro/test/astro-markdown-shiki.test.js b/packages/astro/test/astro-markdown-shiki.test.js
index 88927c4e7..f9c1d0dc4 100644
--- a/packages/astro/test/astro-markdown-shiki.test.js
+++ b/packages/astro/test/astro-markdown-shiki.test.js
@@ -93,11 +93,8 @@ describe('Astro Markdown Shiki', () => {
 			expect(segments[0].attribs.style).to.be.equal('color:#79B8FF');
 			expect(segments[1].attribs.style).to.be.equal('color:#E1E4E8');
 
-			const unknownLang = $('.line').last().html();
-			// TODO: There's a shikiji bug where it's style=""
-			expect(unknownLang).to.be.equal(
-				'<span style="color: #e1e4e8">This language does not exist</span>'
-			);
+			const unknownLang = $('.astro-code').last();
+			expect(unknownLang.attr('style')).to.contain('background-color:#24292e;color:#e1e4e8;');
 		});
 	});