diff --git a/Makefile b/Makefile
index f12211c..e421948 100644
--- a/Makefile
+++ b/Makefile
@@ -4,11 +4,11 @@ all:
 
 .PHONE: test
 test: all
-	tree-sitter test
+	tree-sitter test --apply-all-captures
 
 .PHONE: update-tests
 update-tests: all
-	tree-sitter test -u
+	tree-sitter test -u --apply-all-captures
 
 playground:
 	nix shell pkgs#emscripten --command tree-sitter build-wasm
diff --git a/flake.nix b/flake.nix
index 4c513fb..e7053c4 100644
--- a/flake.nix
+++ b/flake.nix
@@ -93,7 +93,7 @@
               tree-sitter = {
                 enable = true;
                 name = "tree-sitter tests";
-                entry = "${tree-sitter-env}/bin/tree-sitter test";
+                entry = "${tree-sitter-env}/bin/tree-sitter test --apply-all-captures";
                 pass_filenames = false;
               };