diff --git a/lib/character/behaviors/stats.lua b/lib/character/behaviors/stats.lua
index 7a64e98..0c76823 100644
--- a/lib/character/behaviors/stats.lua
+++ b/lib/character/behaviors/stats.lua
@@ -1,7 +1,10 @@
+--- @alias Class "dev_warrior"|"dev_mage"
+
 --- @class StatsBehavior : Behavior
 --- @field hp integer
 --- @field mana integer
 --- @field initiative integer
+--- @field class Class
 --- @field isInTurnOrder boolean
 --- @field amIAlive boolean
 local behavior = {}
@@ -25,14 +28,15 @@ end
 --- @param hp? integer
 --- @param mana? integer
 --- @param initiative? integer
+--- @param class? Class
 --- @param isInTurnOrder? boolean
-function behavior.new(hp, mana, initiative, isInTurnOrder)
+function behavior.new(hp, mana, initiative, class, isInTurnOrder)
     return setmetatable({
         hp = hp or 20,
         mana = mana or 10,
         initiative = initiative or 10,
+        class = class or "dev_warrior",
         isInTurnOrder = isInTurnOrder or true,
-        amIAlive = true
     }, behavior)
 end