diff --git a/layout/_partial/plugins/ai/tianli_gpt.ejs b/layout/_partial/plugins/ai/tianli_gpt.ejs
index 31371ef..8e7ec25 100644
--- a/layout/_partial/plugins/ai/tianli_gpt.ejs
+++ b/layout/_partial/plugins/ai/tianli_gpt.ejs
@@ -1,6 +1,6 @@
 <% if (page.layout === 'post' || (page.wiki?.length > 0)) { %>
   <script>
-    let tianliGPT_postSelector = '.md-text.content.<%= page.layout === "post" ? "post" : "wiki" %>';
+    let tianliGPT_postSelector = 'article.content';
     let tianliGPT_key = '<%= theme.plugins.tianli_gpt.api %>';
     let tianliGPT_typingAnimate = '<%= theme.plugins.tianli_gpt.typingAnimate %>';
     let tianliGPT_wordLimit = <%= theme.plugins.tianli_gpt.limit %>;
diff --git a/layout/page.ejs b/layout/page.ejs
index 6403ae8..12f6b97 100755
--- a/layout/page.ejs
+++ b/layout/page.ejs
@@ -62,6 +62,9 @@ function layoutDiv() {
   if (layout === 'post' || page.wiki) {
     el += partial('_partial/main/article/read_next')
   }
+  if (isUsingTianliGPT) {
+    el += partial('_partial/plugins/ai/tianli_gpt')
+  }
   if (layout === 'post') {
     el += partial('_partial/main/article/related_posts')
   }