diff --git a/queries/smali/highlights.scm b/queries/smali/highlights.scm
index 1e23db7fb..82dd7d758 100644
--- a/queries/smali/highlights.scm
+++ b/queries/smali/highlights.scm
@@ -59,7 +59,6 @@
   ".source"
   ".implements"
   ".field"
-  ".method"
   ".annotation"
   ".subannotation"
   ".param"
@@ -79,12 +78,16 @@
 
 [
   (end_field)
-  (end_method)
   (end_annotation)
   (end_subannotation)
   (end_param)
 ] @keyword
 
+[
+  ".method"
+  (end_method)
+] @keyword.function
+
 ; Literals
 
 (string_literal) @string