diff --git a/lua/neotest-golang/runspec_dir.lua b/lua/neotest-golang/runspec_dir.lua
index d1dc1382..cf02f09d 100644
--- a/lua/neotest-golang/runspec_dir.lua
+++ b/lua/neotest-golang/runspec_dir.lua
@@ -68,7 +68,7 @@ end
 --- @param pos neotest.Position
 --- @param cwd string
 --- @param test_pattern string
---- @return neotest.RunSpec
+--- @return neotest.RunSpec | neotest.RunSpec[] | nil
 function M.build_dir_test_runspec(pos, cwd, test_pattern)
   local gotest = {
     "go",
diff --git a/lua/neotest-golang/runspec_file.lua b/lua/neotest-golang/runspec_file.lua
index 10c72193..ac4373b7 100644
--- a/lua/neotest-golang/runspec_file.lua
+++ b/lua/neotest-golang/runspec_file.lua
@@ -5,7 +5,7 @@ local M = {}
 --- Build runspec for a directory.
 --- @param pos neotest.Position
 --- @param tree neotest.Tree
---- @return neotest.RunSpec | nil
+--- @return neotest.RunSpec | neotest.RunSpec[] | nil
 function M.build(pos, tree)
   if utils.table_is_empty(tree:children()) then
     --- Runspec designed for files that contain no tests.
diff --git a/lua/neotest-golang/runspec_test.lua b/lua/neotest-golang/runspec_test.lua
index 0f9ce87f..9ba34012 100644
--- a/lua/neotest-golang/runspec_test.lua
+++ b/lua/neotest-golang/runspec_test.lua
@@ -6,7 +6,7 @@ local M = {}
 --- Build runspec for a single test
 --- @param pos neotest.Position
 --- @param strategy string
---- @return neotest.RunSpec
+--- @return neotest.RunSpec | neotest.RunSpec[] | nil
 function M.build(pos, strategy)
   --- @type string
   local test_name = convert.to_gotest_test_name(pos.id)