diff --git a/runtime/queries/erlang/highlights.scm b/runtime/queries/erlang/highlights.scm
index cb128905..8e4a640d 100644
--- a/runtime/queries/erlang/highlights.scm
+++ b/runtime/queries/erlang/highlights.scm
@@ -3,7 +3,7 @@
 (attribute
   name: (atom) @keyword
   (arguments (atom) @namespace)
- (#eq? @keyword "module"))
+ (#match? @keyword "(module|behaviou?r)"))
 
 (attribute
   name: (atom) @keyword