diff --git a/queries/python/highlights.scm b/queries/python/highlights.scm
index 84b29412a..bf12846e9 100644
--- a/queries/python/highlights.scm
+++ b/queries/python/highlights.scm
@@ -80,27 +80,10 @@
 ] @string.escape
 
 ; doc-strings
-(module
-  .
-  (comment)*
-  .
-  (expression_statement
-    (string
-      (string_content) @spell) @string.documentation))
+(expression_statement
+  (string
+    (string_content) @spell) @string.documentation)
 
-(class_definition
-  body: (block
-    .
-    (expression_statement
-      (string
-        (string_content) @spell) @string.documentation)))
-
-(function_definition
-  body: (block
-    .
-    (expression_statement
-      (string
-        (string_content) @spell) @string.documentation)))
 
 ; Tokens
 [