diff --git a/queries/python/highlights.scm b/queries/python/highlights.scm
index 5e5a2a88d..b4fa1fc46 100644
--- a/queries/python/highlights.scm
+++ b/queries/python/highlights.scm
@@ -199,46 +199,27 @@
 ] @string.escape
 
 ; doc-strings
-(module
-  .
-  (comment)*
-  .
-  (expression_statement
-    (string) @string.documentation))
-
-(class_definition
-  body: (block
-    .
-    (expression_statement
-      (string) @string.documentation)))
-
-(function_definition
-  body: (block
-    .
-    (expression_statement
-      (string) @string.documentation)))
-
 (module
   .
   (comment)*
   .
   (expression_statement
     (string
-      (string_content) @spell)))
+      (string_content) @spell) @string.documentation))
 
 (class_definition
   body: (block
     .
     (expression_statement
       (string
-        (string_content) @spell))))
+        (string_content) @spell) @string.documentation)))
 
 (function_definition
   body: (block
     .
     (expression_statement
       (string
-        (string_content) @spell))))
+        (string_content) @spell) @string.documentation)))
 
 ; Tokens
 [