diff --git a/queries/python/highlights.scm b/queries/python/highlights.scm
index 329df4f14..0c7147b51 100644
--- a/queries/python/highlights.scm
+++ b/queries/python/highlights.scm
@@ -199,6 +199,8 @@
 
 ; doc-strings
 (module
+  .
+  (comment)*
   .
   (expression_statement
     (string) @string.documentation @spell))