diff --git a/docs/source/_assets/less/markdown.less b/docs/source/_assets/less/markdown.less
index e6f5a64ce..d2a9e8b15 100644
--- a/docs/source/_assets/less/markdown.less
+++ b/docs/source/_assets/less/markdown.less
@@ -113,7 +113,7 @@
     @apply .text-slate-dark;
   }
 
-  > :not(pre) > code[class*="language-"], > pre[class*="language-"], pre[class*="language-"]& {
+  > :not(pre):not(h1):not(h2):not(h3):not(h4) > code, > pre, pre& {
     @apply .bg-smoke-lighter;
     @apply .text-sm;
     @apply .leading-normal;
diff --git a/docs/source/_assets/less/prism.less b/docs/source/_assets/less/prism.less
index bdefa8c06..19d8ffbc3 100644
--- a/docs/source/_assets/less/prism.less
+++ b/docs/source/_assets/less/prism.less
@@ -50,7 +50,7 @@ code[class*="language-"]::selection, code[class*="language-"] ::selection {
 }
 
 /* Code blocks */
-pre[class*="language-"] {
+pre {
     padding: 1em;
     margin: .5em 0;
     overflow: auto;