From 322e4cc65754d2b3fdef4f2f8a71e0762e3d13af Mon Sep 17 00:00:00 2001
From: Tomas Sandven <tomas@sandven.email>
Date: Wed, 16 Aug 2023 22:48:18 +0200
Subject: [PATCH] Code style improvements in queries

---
 queries/highlights.scm | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/queries/highlights.scm b/queries/highlights.scm
index 1d4e414..152526f 100644
--- a/queries/highlights.scm
+++ b/queries/highlights.scm
@@ -6,15 +6,13 @@
 [
   (section_header)
   (setting_statement)
+  (keyword_setting)
+  (test_case_setting)
 ] @keyword
 
 (variable_definition (variable_name) @variable)
-
 (keyword_definition (name) @function)
-(keyword_definition (body (keyword_setting) @keyword))
-
 (test_case_definition (name) @function)
-(test_case_definition (body (test_case_setting) @keyword))
 
 (keyword_invocation (keyword) @function.call)
 (ellipses) @punctuation.delimiter
@@ -35,11 +33,12 @@
   "IN RANGE"
   "IN ENUMERATE"
   "IN ZIP"
-  "WHILE"
   (break_statement)
   (continue_statement)
 ] @repeat
 (for_statement "END" @repeat)
+
+"WHILE" @repeat
 (while_statement "END" @repeat)
 
 [