diff --git a/runtime/queries/python/highlights.scm b/runtime/queries/python/highlights.scm
index 0cdc957d0..0fecd5abd 100644
--- a/runtime/queries/python/highlights.scm
+++ b/runtime/queries/python/highlights.scm
@@ -80,7 +80,11 @@
 ] @string.escape
 
 ; doc-strings
-(expression_statement
+(module
+  (string
+    (string_content) @spell) @string.documentation)
+
+(block
   (string
     (string_content) @spell) @string.documentation)