diff --git a/queries/yaml/folds.scm b/queries/yaml/folds.scm
index f82378e63..c426afb87 100644
--- a/queries/yaml/folds.scm
+++ b/queries/yaml/folds.scm
@@ -1,3 +1,4 @@
 [
-  (block_node)
+  (block_mapping_pair)
+  (block_sequence_item)
 ] @fold