diff --git a/src/entity/gateway/postgres/models.rs b/src/entity/gateway/postgres/models.rs
index 7d508d6..ea8fc21 100644
--- a/src/entity/gateway/postgres/models.rs
+++ b/src/entity/gateway/postgres/models.rs
@@ -384,6 +384,7 @@ impl Into<shield::Shield> for PgShield {
 pub struct PgUnit {
     unit: unit::UnitType,
     modifier: Option<unit::UnitModifier>,
+    armor_slot: u8,
 }
 
 impl From<unit::Unit> for PgUnit {
@@ -391,6 +392,7 @@ impl From<unit::Unit> for PgUnit {
         PgUnit {
             unit: other.unit,
             modifier: other.modifier,
+            armor_slot: other.armor_slot,
         }
     }
 }
@@ -400,6 +402,7 @@ impl Into<unit::Unit> for PgUnit {
         unit::Unit {
             unit: self.unit,
             modifier: self.modifier,
+            armor_slot: self.armor_slot,
         }
     }
 }