diff --git a/static/css/extensions/jupyter.css b/static/css/extensions/jupyter.css
index 6f0c896..e700149 100644
--- a/static/css/extensions/jupyter.css
+++ b/static/css/extensions/jupyter.css
@@ -17,6 +17,10 @@
   padding: 0;
 }
 
+.yue .jupyter_container div.highlight {
+  background-color: var(--syntax-pre-bg);
+}
+
 .yue .jupyter_container div.cell_input {
   background-color: var(--syntax-pre-bg);
   border: 0;