diff --git a/queries/tlaplus/highlights.scm b/queries/tlaplus/highlights.scm
index f5dd93ac2..bbbabd968 100644
--- a/queries/tlaplus/highlights.scm
+++ b/queries/tlaplus/highlights.scm
@@ -66,7 +66,7 @@
   (case_box)
   (case_arrow)
   (label_as)
-] @number
+] @keyword
 [
   (def_eq)
   (maps_to)