diff --git a/lua/cp/cache.lua b/lua/cp/cache.lua
index ebf506f..f1f2f8f 100644
--- a/lua/cp/cache.lua
+++ b/lua/cp/cache.lua
@@ -146,7 +146,7 @@ end
 ---@param platform string
 ---@param contest_id string
 ---@param problem_id? string
----@return TestCase[]?
+---@return CachedTestCase[]?
 function M.get_test_cases(platform, contest_id, problem_id)
 	vim.validate({
 		platform = { platform, "string" },
@@ -164,7 +164,7 @@ end
 ---@param platform string
 ---@param contest_id string
 ---@param problem_id? string
----@param test_cases TestCase[]
+---@param test_cases CachedTestCase[]
 function M.set_test_cases(platform, contest_id, problem_id, test_cases)
 	vim.validate({
 		platform = { platform, "string" },