diff --git a/src/extension.ts b/src/extension.ts
index 03a12bc..676f178 100644
--- a/src/extension.ts
+++ b/src/extension.ts
@@ -15,8 +15,8 @@ import {
     TextEditorEdit
 } from 'vscode';
 
-const REGEXP_TOC_START          = /\s*<!--(.*)TOC(.*)-->/gi;
-const REGEXP_TOC_STOP           = /\s*<!--(.*)\/TOC(.*)-->/gi;
+const REGEXP_TOC_START          = /\s*<!--\s*TOC(\s(\.*))?-->/gi;
+const REGEXP_TOC_STOP           = /\s*<!--\s\/TOC(\s(\.*))?-->/gi;
 const REGEXP_TOC_CONFIG         = /\w+[:=][\w.]+/gi;
 const REGEXP_TOC_CONFIG_ITEM    = /(\w+)[:=]([\w.]+)/;
 const REGEXP_MARKDOWN_ANCHOR    = /^<a id="markdown-.+" name=".+"><\/a\>/;