diff --git a/media/lua/client/TOC/Handlers/ModDataHandler.lua b/media/lua/client/TOC/Handlers/ModDataHandler.lua
index face0b6..8abeb39 100644
--- a/media/lua/client/TOC/Handlers/ModDataHandler.lua
+++ b/media/lua/client/TOC/Handlers/ModDataHandler.lua
@@ -67,12 +67,14 @@ function ModDataHandler:setup(key)
     end
 
     -- Initialize prostheses stuff
-    for group, _ in ipairs(StaticData.PROSTHESES_GROUPS) do
+    -- TODO This is shit
+    local groups = {"top", "bottom"}
+    for i=1, #groups do
+        local group = groups[i]
         self.tocData.prostheses[group] = {
             isEquipped = false,
-            prostFactor = 0
+            prostFactor = 0,
         }
-        -- TODO Make this more modular instead of this crap
     end
 
     -- Add it to global mod data
@@ -113,9 +115,9 @@ end
 
 ---Set isProstEquipped
 ---@param group string
----@param isProstEquipped boolean
-function ModDataHandler:setIsProstEquipped(group, isProstEquipped)
-    self.tocData.prostheses[group].isProstEquipped = isProstEquipped
+---@param isEquipped boolean
+function ModDataHandler:setIsProstEquipped(group, isEquipped)
+    self.tocData.prostheses[group].isEquipped = isEquipped
 end
 
 ---Set prostFactor
diff --git a/media/lua/shared/TOC/StaticData.lua b/media/lua/shared/TOC/StaticData.lua
index f24dc63..f88f6cd 100644
--- a/media/lua/shared/TOC/StaticData.lua
+++ b/media/lua/shared/TOC/StaticData.lua
@@ -1,7 +1,7 @@
 ---@alias partData { isCut : boolean?, isInfected : boolean?, isOperated : boolean?, isCicatrized : boolean?, isCauterized : boolean?, isVisible : boolean?, cicatrizationTime : number }
 ---@alias limbsTable {Hand_L : partData, ForeArm_L : partData, UpperArm_L : partData, Hand_R : partData, ForeArm_R : partData, UpperArm_R : partData }
 ---@alias prosthesisData {isEquipped : boolean, prostFactor : number }
----@alias prosthesesTable {top : table, bottom : table }
+---@alias prosthesesTable {top : prosthesisData, bottom : prosthesisData }
 ---@alias tocModData { limbs : limbsTable, prostheses : prosthesesTable, isIgnoredPartInfected : boolean, isAnyLimbCut : boolean }
 ---------------------------