diff --git a/queries/elixir/highlights.scm b/queries/elixir/highlights.scm
index 1a93a7687..2553ba6c9 100644
--- a/queries/elixir/highlights.scm
+++ b/queries/elixir/highlights.scm
@@ -209,7 +209,7 @@
 (unary_operator
   operator: "@"
   operand: (call
-    target: ((identifier) @_identifier (#match? @_identifier "doc$")) @comment
+    target: ((identifier) @_identifier (#any-of? @_identifier "moduledoc" "typedoc" "shortdoc" "doc")) @comment
     (arguments [
       (string)
       (boolean)