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