diff --git a/queries/highlights.scm b/queries/highlights.scm
index b66fdbe..0af3290 100644
--- a/queries/highlights.scm
+++ b/queries/highlights.scm
@@ -2,7 +2,8 @@
 (quoted_element) @parameter
 (unquoted_argument) @parameter
 (variable) @variable.builtin
-(command_invocation [
+[
   (normal_command)
-  (foreach_loop)
-] @function)
+  (foreach_command)
+  (endforeach_command)
+]@function