diff --git a/queries/python/fold.scm b/queries/python/fold.scm
index 16c0cb482..58ce4772c 100644
--- a/queries/python/fold.scm
+++ b/queries/python/fold.scm
@@ -1,13 +1,13 @@
-(function_definition (block) @fold)
-(class_definition (block) @fold)
-
-(while_statement (block) @fold)
-(for_statement (block) @fold)
-(if_statement (block) @fold)
-(with_statement (block) @fold)
-(try_statement (block) @fold)
-
 [
+  (function_definition)
+  (class_definition)
+
+  (while_statement)
+  (for_statement)
+  (if_statement)
+  (with_statement)
+  (try_statement)
+
   (import_from_statement)
   (parameters)
   (argument_list)