diff --git a/tests/query/highlights/proto/test.proto b/tests/query/highlights/proto/test.proto
index fdc610782..f037d8ef3 100644
--- a/tests/query/highlights/proto/test.proto
+++ b/tests/query/highlights/proto/test.proto
@@ -1,11 +1,11 @@
 syntax = "proto2";
-//       ^^^^^^^^ @string.special
+//       ^^^^^^^^ @string.special.symbol
 package sls.asfd.asfd;
-//      ^^^ @variable
+//      ^^^ @module
 //         ^ @punctuation.delimiter
-//          ^^^^ @variable.member
+//          ^^^^ @module
 //              ^ @punctuation.delimiter
-//               ^^^^ @variable.member
+//               ^^^^ @module
 //                   ^ @punctuation.delimiter
 
 import "github.com/gogo/protobuf/gogoproto/gogo.proto";
@@ -21,10 +21,6 @@ import "github.com/gogo/protobuf/gogoproto/gogo.proto";
 option (gogoproto.sizer_all) = true;
 option (gogoproto.marshaler_all) = true;
 option (gogoproto.unmarshaler_all.foo) = true;
-//      ^^^^^^^^^ @variable
-//                ^^^^^^^^^^^^^^^ @variable.member
-//                               ^ @punctuation.delimiter
-//                                ^^^ @variable.member
 
 message LogContent
 //<- @keyword.type
@@ -46,7 +42,7 @@ message LogContent
     optional JsonFormat json_format = 6 [
 //  ^^^^^^^^ @keyword.modifier
       retention = RETENTION_RUNTIME,
-//                ^^^^^^^^^^^^^^^^^ @variable
+//                ^^^^^^^^^^^^^^^^^ @constant
 //                                 ^ @punctuation.delimiter
       targets = TARGET_TYPE_MESSAGE,
 //    ^^^^^^^ @property
@@ -60,3 +56,4 @@ message LogContent
     ];
 }
 
+