diff --git a/runtime/lua/vim/treesitter/_select.lua b/runtime/lua/vim/treesitter/_select.lua
index 90c83e1aa8..2fb5fc1561 100644
--- a/runtime/lua/vim/treesitter/_select.lua
+++ b/runtime/lua/vim/treesitter/_select.lua
@@ -364,7 +364,7 @@ local function visual_select(range)
     ecol = #vim.fn.getline(erow + 1) + 1
   end
 
-  if vim.api.nvim_get_mode().mode ~= 'v' then
+  if vim.fn.visualmode() ~= 'v' then
     -- reset visualmode() to 'v'
     vim.cmd.normal({ 'v\27', bang = true })
   end