From 9f12ad1142815c0d9078fe54b382ca5650ba8756 Mon Sep 17 00:00:00 2001 From: Stephan Seitz Date: Mon, 16 Nov 2020 21:22:52 +0100 Subject: Fix "nth?" predicate --- lua/nvim-treesitter/query_predicates.lua | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'lua') diff --git a/lua/nvim-treesitter/query_predicates.lua b/lua/nvim-treesitter/query_predicates.lua index d5aacb1ec..a41ad287b 100644 --- a/lua/nvim-treesitter/query_predicates.lua +++ b/lua/nvim-treesitter/query_predicates.lua @@ -24,9 +24,9 @@ query.add_predicate("nth?", function(match, pattern, bufnr, pred) if not valid_args("nth?", pred, 2, true) then return end local node = match[pred[2]] - local n = pred[3] - 1 - if node and node:parent() and node:named_child_count() > n then - return node:named_child(n) == node + local n = pred[3] + if node and node:parent() and node:parent():named_child_count() > n then + return node:parent():named_child(n) == node end return false -- cgit v1.2.3-70-g09d2