diff --git a/static/koboldai.css b/static/koboldai.css
index 188b1be2..f3c17415 100644
--- a/static/koboldai.css
+++ b/static/koboldai.css
@@ -1627,7 +1627,7 @@ body {
 	opacity: 0.5;
 }
 
-.context-block {
+.context-block, .context-block-example{
 	margin: 0px 2px;
 }
 
@@ -1635,10 +1635,6 @@ body {
 	outline: 1px solid gray;
 }
 
-.context-block.example:hover {
-	outline: none !important;
-}
-
 .context-sp {background-color: orangered;}
 .context-prompt {background-color: orange;}
 .context-wi {background-color: #1751b9;}
diff --git a/static/koboldai.js b/static/koboldai.js
index 87729823..290f4667 100644
--- a/static/koboldai.js
+++ b/static/koboldai.js
@@ -2044,10 +2044,7 @@ function update_bias_slider_value(slider) {
 }
 
 function update_context(data) {
-	for (const el of document.getElementsByClassName("context-block")) {
-		if (el.classList.contains("example")) continue;
-		el.remove();
-	}
+	$(".context-block").remove();
 
 	for (const entry of data) {
 		console.log(entry);
diff --git a/templates/index_new.html b/templates/index_new.html
index 89675cee..f21e54f8 100644
--- a/templates/index_new.html
+++ b/templates/index_new.html
@@ -121,12 +121,12 @@
 					<div>
 						<span class="noselect">Key:</span>
 						<div>
-							<span class="noselect example context-block context-sp">Soft Prompt</span>
-							<span class="noselect example context-block context-prompt">Prompt</span>
-							<span class="noselect example context-block context-wi">World Info</span>
-							<span class="noselect example context-block context-memory">Memory</span>
-							<span class="noselect example context-block context-an">Author's Note</span>
-							<span class="noselect example context-block context-action">Action</span>
+							<span class="noselect context-block-example context-sp">Soft Prompt</span>
+							<span class="noselect context-block-example context-prompt">Prompt</span>
+							<span class="noselect context-block-example context-wi">World Info</span>
+							<span class="noselect context-block-example context-memory">Memory</span>
+							<span class="noselect context-block-example context-an">Author's Note</span>
+							<span class="noselect context-block-example context-action">Action</span>
 						</div>
 					</div>
 					<span id="context-viewer-close" class="material-icons-outlined">close</span>