diff --git a/tests/query/highlights/kos/test.kos b/tests/query/highlights/kos/test.kos
index cffa21218..6ab57ebca 100644
--- a/tests/query/highlights/kos/test.kos
+++ b/tests/query/highlights/kos/test.kos
@@ -9,13 +9,13 @@ public var item = 1;
 #                  ^ punctuation.delimiter
 
 # hash comment
-# <- spell
+# <- @comment @spell
 
 // slash comment
-# <- spell
+# <- @comment @spell
 
 /* block comment */
-# <- spell
+# <- @comment @spell
 
 fun name(arg1,
 # ^ keyword.function