diff --git a/crates/languages/src/python/highlights.scm b/crates/languages/src/python/highlights.scm
index 6c3f027c1976597c82a4fbe1d86f0bbe8d23357b..98ed2039695c7e2fffaf6a9e1a9e8d948b8e517a 100644
--- a/crates/languages/src/python/highlights.scm
+++ b/crates/languages/src/python/highlights.scm
@@ -96,7 +96,16 @@
   "def"
   name: (_)
   (parameters)?
-  body: (block (expression_statement (string) @string.doc)))
+  body: (block . (expression_statement (string) @string.doc)))
+
+(class_definition
+  body: (block
+    . (comment) @comment*
+    . (expression_statement (string) @string.doc)))
+
+(module
+  . (comment) @comment*
+  . (expression_statement (string) @string.doc))
 
 (module
   (expression_statement (assignment))