diff --git a/themes/init.js b/themes/init.js
index f240d80e..06cb5616 100644
--- a/themes/init.js
+++ b/themes/init.js
@@ -250,10 +250,16 @@ async function GptAcademicJavaScriptInit(dark, prompt, live2d, layout, tts) {
         searchString = "副标题";
         tool_bar_group = "cbsc";
         const true_function = function () {
-            document.querySelector('.prose.svelte-1ybaih5 h2').style.display = '';
+            element = document.querySelector('.prose.svelte-1ybaih5 h2');
+            if (element) element.style.display = '';
+            element = document.querySelector('.prose.svelte-1ybaih5 h6');
+            if (element) element.style.display = '';
         }
         const false_function = function () {
-            document.querySelector('.prose.svelte-1ybaih5 h2').style.display = 'none';
+            element = document.querySelector('.prose.svelte-1ybaih5 h2');
+            if (element) element.style.display = 'none';
+            element = document.querySelector('.prose.svelte-1ybaih5 h6');
+            if (element) element.style.display = 'none';
         }
         if (bool_value) {
             // make btns appear