diff --git a/queries/cmake/highlights.scm b/queries/cmake/highlights.scm
index 055319978..80bf80752 100644
--- a/queries/cmake/highlights.scm
+++ b/queries/cmake/highlights.scm
@@ -1,7 +1,7 @@
 (normal_command
   (identifier)
   (argument (unquoted_argument)) @constant
-  (#lua-match? @constant "^%u[%u_]+$")
+  (#lua-match? @constant "^[%u@][%u%d_]+$")
 )
 
 [