diff --git a/lib/extend/injector.ts b/lib/extend/injector.ts
index 0f341f519b..06485e829c 100644
--- a/lib/extend/injector.ts
+++ b/lib/extend/injector.ts
@@ -66,14 +66,7 @@ class Injector {
   }
 
   _injector(input: string, pattern: string | RegExp, flag: Entry, isBegin = true, currentType: string): string {
-    if (
-      (flag==='head_begin' && /<head.*?>\s*<!-- hexo injector head_begin start -->/.test(input)) ||
-      (flag==='body_begin' && /<body.*?>\s*<!-- hexo injector body_begin start -->/.test(input)) ||
-      (flag==='head_end' && /<!-- hexo injector head_end end -->\s*<\/head>/.test(input)) ||
-      (flag==='body_end' && /<!-- hexo injector body_end end -->\s*<\/body>/.test(input))
-    ) {
-      return input;
-    }
+    if (input.includes(`<!-- hexo injector ${flag}`)) return input;
 
     const code = this.cache.apply(`${flag}-${currentType}-code`, () => {
       const content = currentType === 'default' ? this.getText(flag, 'default') : this.getText(flag, currentType) + this.getText(flag, 'default');
diff --git a/test/scripts/extend/injector.ts b/test/scripts/extend/injector.ts
index 24d7c2106e..cb9ff38e83 100644
--- a/test/scripts/extend/injector.ts
+++ b/test/scripts/extend/injector.ts
@@ -167,6 +167,7 @@ describe('Injector', () => {
     result.should.contain('<body id="body"><!-- hexo injector body_begin start --><!-- hexo injector body_begin end -->');
     result.should.contain('<!-- hexo injector body_end start --><script src="prism.js"></script><!-- hexo injector body_end end --></body>');
   });
+  
 
   it('exec() - multi-line head & body', () => {
     const content = [
