diff --git a/lib/character/behaviors/stats.lua b/lib/character/behaviors/stats.lua
index 965d290..c3bb08e 100644
--- a/lib/character/behaviors/stats.lua
+++ b/lib/character/behaviors/stats.lua
@@ -1,3 +1,5 @@
+--- @alias Class "dev_warrior"|"dev_mage"
+
 --- @class StatsBehavior : Behavior
 --- @field hp integer
 --- @field mana integer
@@ -10,13 +12,15 @@ behavior.id = "stats"
 --- @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,
-        isInTurnOrder = isInTurnOrder or true
+        class = class or "dev_warrior",
+        isInTurnOrder = isInTurnOrder or true,
     }, behavior)
 end