diff --git a/lua/neotest/utils/init.lua b/lua/neotest/utils/init.lua
index 0423418..95ed42a 100644
--- a/lua/neotest/utils/init.lua
+++ b/lua/neotest/utils/init.lua
@@ -1,7 +1,9 @@
+local nio = require("nio")
+
 local M = {}
 
 function M.tbl_flatten(t)
-  return vim.fn.has("nvim-0.11") == 1 and vim.iter(t):flatten(math.huge):totable()
+  return nio.fn.has("nvim-0.11") == 1 and vim.iter(t):flatten(math.huge):totable()
     or vim.tbl_flatten(t)
 end