diff --git a/runtime/queries/python/highlights.scm b/runtime/queries/python/highlights.scm
index 35294072..93e08d58 100644
--- a/runtime/queries/python/highlights.scm
+++ b/runtime/queries/python/highlights.scm
@@ -34,6 +34,9 @@
 
 (decorator) @function
 (decorator (identifier) @function)
+(decorator (attribute attribute: (identifier) @function))
+(decorator (call
+  function: (attribute attribute: (identifier) @function)))
 
 ; Parameters