diff --git a/src/Unit/Unit.lua b/src/Unit/Unit.lua
index 747c041..9e8ddf4 100644
--- a/src/Unit/Unit.lua
+++ b/src/Unit/Unit.lua
@@ -528,6 +528,7 @@ function Unit:IsMovingAtAll()
     return ObjectMovementFlag(self:GetOMToken()) ~= 0
 end
 
+---@param unit Unit | nil
 ---@return number
 function Unit:GetComboPoints(unit)
     if Tinkr.classic then
@@ -548,6 +549,7 @@ function Unit:GetComboPointsMax()
 end
 
 -- Get combopoints deficit
+---@param unit Unit | nil
 ---@return number
 function Unit:GetComboPointsDeficit(unit)
     if Tinkr.classic then