diff --git a/lua/resession/init.lua b/lua/resession/init.lua
index 7a51ec5..06c1c3f 100644
--- a/lua/resession/init.lua
+++ b/lua/resession/init.lua
@@ -622,14 +622,14 @@ end
 
 ---Add a callback that runs at a specific time
 ---@param name "pre_save"|"post_save"|"pre_load"|"post_load"
----@param callback fun()
+---@param callback fun(...)
 M.add_hook = function(name, callback)
   table.insert(hooks[name], callback)
 end
 
 ---Remove a hook callback
 ---@param name "pre_save"|"post_save"|"pre_load"|"post_load"
----@param callback fun()
+---@param callback fun(...)
 M.remove_hook = function(name, callback)
   local cbs = hooks[name]
   for i, cb in ipairs(cbs) do