diff --git a/queries/proto/highlights.scm b/queries/proto/highlights.scm
index a6f47da93..a24314834 100644
--- a/queries/proto/highlights.scm
+++ b/queries/proto/highlights.scm
@@ -1,8 +1,6 @@
 [
   "syntax"
-  "package"
   "option"
-  "import"
   "service"
   "rpc"
   "returns"
@@ -15,6 +13,11 @@
   "to"
 ] @keyword
 
+[
+  "package"
+  "import"
+] @include
+
 [
   (key_type)
   (type)