1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13

diff --git a/src/extension.ts b/src/extension.ts
index 9127c1e0519c08b828bcbdc6b31d6af17d7c4366..1d060ecd1e87703bd79fd00758fa6e40de94e92a 100644
--- a/src/extension.ts
+++ b/src/extension.ts
@@ -380,7 +380,7 @@             position
           )
           const text1: string = document.getText(range1)
 
-          if (!/class=['"][^'"]*$/.test(text1)) return
+          if (!/\bclass(Name)?=['"][^'"]*$/.test(text1)) return
 
           const range2: vscode.Range = new vscode.Range(
             new vscode.Position(Math.max(position.line - 5, 0), 0),