diff --git a/queries/lua/highlights.scm b/queries/lua/highlights.scm
index 507d21989..439f570bd 100644
--- a/queries/lua/highlights.scm
+++ b/queries/lua/highlights.scm
@@ -101,12 +101,12 @@
 
 ;; Functions
 (function [(function_name) (identifier)] @function)
-(function ["function" "end"] @function)
+(function ["function" "end"] @keyword.function)
 
 (local_function [(function_name) (identifier)] @function)
-(local_function ["function" "end"] @function)
+(local_function ["function" "end"] @keyword.function)
 
-(function_definition ["function" "end"] @function)
+(function_definition ["function" "end"] @keyword.function)
 
 (property_identifier) @property
 (method) @method